在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch

conda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch

参考:https://blog.csdn.net/qq_37774098/article/details/109895048
https://bbs.cvmart.net/topics/3514

操作实例

显卡驱动安装适合本机显卡的最新版本。

cuda安装

1 确定cuda版本

安装好显卡驱动后,使用nvidia-smi命令可以查看这个显卡驱动可以安装的最高的cuda版本是多少,如下:
Driver Version: 461.40 表明当前驱动版本是461.40
CUDA Version: 11.2 表明当前驱动可以安装的cuda最高版本是11.2

在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch
上面虽然显示的当前驱动版本461.40可以安装的cuda最高版本是11.2,但是我们还不能直接就认定我们就安装cuda 11.2,还需要查看pytorch的版本要求,比如我想安装pytorchv1.8.0,如下:
在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch
pytorchv1.8.0 要求的cuda版本是11.1,那我就可以确定安装cuda11.1

; 2 在conda环境中安装cuda

conda search cudatoolkit

conda search cudatoolkit --info

在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch

conda install --use-local H:\装机软件\deeplearningsoftware\cudatoolkit-11.1.1-heb2d755_7.tar.bz2

cudnn安装

确定cudnn版本

在cudnn仓库中找到适用于某个cudatoolkit版本的cudnn版本


https://developer.nvidia.com/rdp/cudnn-archive

在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch
适用于CUDA 11.1 的 cudnn 有多个版本。
查看conda仓库中有哪些cudnn版本:

conda search cudnn

在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch

conda search cudnn --info

在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch

确定tensorflow对应的cuda和cudnn版本

在tensorflow官网——安装——cuda支持可以查看

在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch
在此处有关于cuda和cudnn的版本支持说明,请注意这句话 这些安装说明适用于最新版 TensorFlow。如需了解可用于旧版 TensorFlow 的 CUDA® 和 cuDNN 版本,请参阅经过测试的构建配置
在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch
经过测试的构建配置链接中包含如下说明表:下面仅展示了linux下的,当然也有windows下的,这里就不展示了
在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch
回到上面那句话这句话 这些安装说明适用于最新版 TensorFlow。如需了解可用于旧版 TensorFlow 的 CUDA® 和 cuDNN 版本,请参阅经过测试的构建配置。我们来查看 这些安装说明适用于最新版 TensorFlow。下面说明了,TensorFlow 支持 CUDA® 11.2(TensorFlow 2.5.0 及更高版本), CUDA® 11.2 要求 450.80.02 或更高版本,cuDNN SDK 8.1.0 cuDNN 版本。
在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch

; 注意1

在conda环境中,可以直接用conda装cuda和cudnn,缺点是conda源中没有全部版本的cuda和cudnn,而且更新较慢。


conda search cuda
conda search cudnn

conda install cuda
conda install cudnn

注意2

和安装tensorflow不同的是,pytorch官网介绍的安装命令附带了下载和安装cuda,但是没有附带cudnn,所以还需要额外安装cudnn。
如下:

在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch
按照上面的命令安装好(不用conda安装cudnn)之后,使用网上的”检测pytorch的gpu是否可用的命令”检测时显示gpu可用,但是没有cudnn,所以还是不能用。

https://blog.csdn.net/weixin_42788078/article/details/103116903
https://www.jianshu.com/p/bdb91eeb8f16

所以要用conda命令安装cudnn

https://blog.csdn.net/qq_37774098/article/details/109895048

安装torch和torchvision时候,使用pip命令,不要用conda命令,很容易断掉

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

注意3 conda虚拟环境安装CUDA路径


https://blog.csdn.net/Mr__George/article/details/102972958

注意4 conda install临时用国内源

conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.1.4 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

Windows下使用:
conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/

注意5 conda install加上官方默认源


conda install -c conda-forge  cudatoolkit=9.0

Original: https://blog.csdn.net/LIWEI940638093/article/details/113811563
Author: xianglingliwei
Title: 在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow(-gpu)、pytorch

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/522124/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球