Ubuntu18.04+CUDA10.1+cudnn7.6.5+tensorflow1.14+pytorch1.4安装全过程

1、安装NVIDIA驱动

2、安装CUDA10.1

3、安装cudnn7.6.5

4、安装tensorflow1.14

5、安装pytorch1.4

 

1、安装NVIDIA驱动：

https://blog.csdn.net/qq_31456593/article/details/90170708  主要按照这个来安装

1.  通过 ubuntu-drivers devices 命令查看推荐的驱动的版本 （我的推荐的是440）
2. 去 https://www.geforce.cn/drivers下载相应的驱动
3. sudo chmod 777 NVIDIA-Linux-x86_64-440.100.run
4. sudo ./NVIDIA-Linux-x86_64-410.100.run
5. # 查看显卡驱动 nvidia-smi

 

2、安装CUDA10.1：

https://blog.csdn.net/silver1225/article/details/100393719 主要按照这个来安装

1.   https://developer.nvidia.com/cuda-toolkit-archive  去cuda的历史版本下载cuda10.1 下载runfile
2.  sudo apt-get install gcc g++ make  安装其他需要的依赖工具
3.  sudo sh cuda_10.1.105_418.39_linux.run 
4. 
5.  
6. 
7.  

            这个选择不安装驱动，因为第一步已经装了。还有下一步要更新 选择第二个 全部更新 

    后面测试的样例就按照链接中来

 

3、安装cudnn7.6.5：

https://blog.csdn.net/silver1225/article/details/100393719 主要按照这个来安装

1.  由于我注册不了NVIDIA账号，所以我就用别人分享的CUDNN来安装
2. 链接： https://pan.baidu.com/s/1P6CIGMK2t4ffF2tk2BS6YA
3. 提取码：vaiz
4. 在按照上面的链接一步一步安装测试。

 

4、安装tensorflow1.14：

        pip install tensorflow-gpu=1.14 -i http://pypi.douban.com/simple/  --trusted-host pypi.douban.com

 

5、安装pytorch1.4：

        https://download.pytorch.org/whl/torch_stable.html

        去这个链接下载历史版本 

        pip install  XXXXX.whl就可以了