安装 cuda 10.1 与 cudnn7.6.5、TensorFlow-GPU 2.2 与pytorch-GPU 1.5

目录

文章汇总

一、下载 cuda 与 cudnn

二、安装 cuda 与 cudnn

三、安装 TensorFlow 与 pytorch

---

文章汇总

           win10与Ubuntu16.04双系统安装、配置 Anaconda3环境、配置cuda与cudnn、配置TensorFlow-gpu 与 pytorch-gpu、ssh内网穿透

           tip：文章中有网络图片，实际情况以自己电脑为准

所有的文件下载网盘：
链接: https://pan.baidu.com/s/1uVeter5GWBG_qYqH2qVvbA  密码: qs5w
--来自百度网盘超级会员V4的分享

一、下载 cuda 与 cudnn

1. Ubuntu cuda、cudnn 版本：cuda 10.1、cudnn 7.6.5 for cuda 10.1 【cuda官网、cudnn官网 [ 需要注册 ]、cuda 与 cudnn TensorFlow 对照表官网】

          tip：cuda 下载  .deb 格式，  cudnn 下载  library 包

     

二、安装 cuda 与 cudnn

      参考视频：B站 linux 深度学习环境搭建

      几个常见的问题回答：

1、新装的Ubuntu系统没有NVIDIA的驱动【前提是在安装ubantu系统是没有点击安装第三方程序】，那么需要先安装NVIDIA
   驱动在安装cuda吗？
2、已经安装过的NVIDIA驱动怎么卸载？
3、我怎么单独安装NVIDIA驱动？
4、conda 命令在安装 TensorFlow-gpu时会自动安装 cuda 与 cudnn，为什么手动安装？ 

回答：
1、不需要，我们在官网下载cuda的时候优先选择 .deb 格式，因为这个格式的文件在安装cuda的时候也会为您的计算机安装NVIDIA驱动。
   tip：事先提醒一下，这个过程下载的NVIDIA驱动是从外网下载的，你懂得，建议网速不好或者没有代理科学上网的童鞋做好下载
       一晚上的时间的准备

回答：
2、卸载已安装的NVIDIA驱动可以参考下面的一篇文章：
    https://blog.csdn.net/ezhchai/article/details/80536949

回答：
3、单独安装 NVIDIA 驱动（.run格式）参考以下文章： 

          最详细ubuntu16.04安装nvidia显卡驱动 【这篇足够了，但是有个缺点，可能会自动安装一个较高版本的 cuda，所以建议还是用 cuda . deb 格式安装就好 】

回答：
4、个人习惯吧，anaconda base 环境下手动安装 cuda 与 cudnn 并配置环境变量，方便使用 nvidia-smi 与 nvcc -V 命令，
   如果使用conda管理多个TensorFlow版本时，可以使用 conda 安装TensorFlow，这个时候conda会自动安装仅仅使用该环境下的 
   cuda 与 cudnn，不会与环境变量中的 cuda 与 cudnn 冲突

三、安装 TensorFlow 与 pytorch

1. Ubuntu TensorFlow-gpu 版本：TensorFlow-gpu==2.2 【手动安装 cuda 或 cudnn 后，使用 pip 安装】 
2. Ubuntu pytorch-gpu 版本：pytorch-gpu=1.5   【手动安装 cuda 或 cudnn 后，使用 pip 安装】
3. cuda 与 cudnn TensorFlow 对照表官网 ，pytorch 与 cuda、cudnn 也是有对照表的【pytorch官网】

# 因为已经手动安装过cuda 与 cudnn了，使用阿里云镜像pip秒装即可
pip install -i https://mirrors.aliyun.com/pypi/simple/ --upgrade tensorflow-gpu==2.2
pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html