ubuntu18.4+cuda10.0+CUDnn7.6+tensorflow1.14+pytorch1.2深度学习环境搭建过程与遇到的问题总结

1.ubuntu18.4

我是在window10系統基礎上安裝的ubuntu18.4的双系统，具体安装过程请见另一篇文章

2.显卡驱动 435.21

2.1安装

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get updata
sudo apt-get update 

到软件和更新的附加驱动选项卡，选择专有驱动，应用更改
到终端输入命令：

 nvidia-smi

3.CUDA 10.0.130

3.1 安装

如何选择cuda版本

降级gcc版本

sudo apt-get install  gcc-6  g++-6
cd /usr/bin/
sudo rm gcc
sudo ln -s gcc-6 gcc
sudo rm g++
sudo ln -s g++-6 g++

查看gcc 版本

 gcc --verson

sudo sh cuda_10.0.130_410.48_linux.run
gedit  ~/.bashrc
source  ~/.bashrc

4.CUDnn 7.6.4

4.1安装

如何选择版本

tar -zxvf cudnn-10.0-linux-x64-v7.6.4.38.tgz

 sudo cp cuda/include/cudnn.h  /usr/local/cuda/include/
 sudo cp cuda/lib64/libcudnn*  /usr/local/cuda/lib64/
 sudo chmod a+r /usr/local/cuda/include/cudnn.h  /usr/local/cuda/lib64/libcudnn*

5.Anaconda3

5.1安装

 bash Anaconda3-2020.02-Linux-x86_64.sh 

python

 vim /etc/profile

/home/david/anaconda3/bin/

 source /etc/profile 

6.tensorflow-gpu

6.1 安装

更新conda源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

更新pip元

mkdir ~/.pip

cat > ~/.pip/pip.conf << EOF
#加入
[global]
trusted-host=mirrors.aliyun.com
index-url=https://mirrors.aliyun.com/pypi/simple/
EOF

安装tensorflow
怎么选择版本

 pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.14.0

问题：

解决：降级numpy 1.17版本以下
卸载numpy

 pip uninstall numpy

安装指定版本numpy

 pip install numpy==1.16

7.pytorch1.2

7.1安装

寻找对应版本，pytorch是facebook开发的一款深度学习框架，比tensorflow1.0版本更容易学习，易操作，深受喜爱；
当google发布TensorFlow 2.0，TensorFlow也变的容易使用

7.2使用遇到的问题：

问题一：
通过conda安装pytorch完成之后， import torchvision时候报错

解决一：

ImportError: cannot import name 'PILLOW_VERSION' from 'PIL' (/home/david/anaconda3/lib/python3.7/site-packages/PIL/__init__.py)

查看/home/david/anaconda3/lib/python3.7/site-packages/PIL/init.py文件，主要相关内容如下：

from . import _version

# VERSION was removed in Pillow 6.0.0.
# PILLOW_VERSION was removed in Pillow 7.0.0.
# Use __version__ instead.
__version__ = _version.__version__

del _version

我们可以发现，PILLOW_VERSION已在7.0版本不存在了，可以使用__version__替换，于是找到报错上一行代码：

 File "/home/david/anaconda3/lib/python3.7/site-packages/torchvision/transforms/functional.py", line 5, in <module>
    from PIL import Image, ImageOps, ImageEnhance, PILLOW_VERSION

对文件/home/david/anaconda3/lib/python3.7/site-packages/torchvision/transforms/functional.py进行修改：

把文件中代码

from PIL import Image, ImageOps, ImageEnhance, PILLOW_VERSION

替换为：

from PIL import Image, ImageOps, ImageEnhance, __version__

保存即可。