Anaconda直接使用conda命令安装cuda和cudnn

Anaconda直接使用conda命令安装cuda和cudnn

  • 背景
  • 一、确定所需要的cuda版本
    • tesorflow对应cuda版本查询
    • pytorch对应cuda版本查询
  • 二、安装指令
    • 安装cuda
    • 安装cudnn

背景

之前在电脑上配置cuda和cudnn,那一顿操作,各种下载,复杂不说,还不能随意更难改,不能和各个项目良好对应。知道的很晚,可以直接通过虚拟环境安装cuda和cudnn,两行代码,超级方便,各个环境下有自己的cuda和cudnn,不会相互干扰,实在是很方便了。

一、确定所需要的cuda版本

tesorflow对应cuda版本查询

tensorflow对应的cuda版本
例:
Anaconda直接使用conda命令安装cuda和cudnn_第1张图片

pytorch对应cuda版本查询

执行conda install pytorch=X.X.X -c pytorch时,conda会自动为你选择合适版本的cudatoolkit。但conda只能保证你的pytorch和cudatoolkit版本一定是对应的,但并不能保证pytorch可以正常使用,因为系统的Nvidia Driver有可能不支持你所安装的cudatoolkit版本。

所以,除非你对你的Nvidia driver版本很有自信,否则,还是先查看系统Nvidia Driver的版本,并在上方图表中查询最高支持的cudatoolkit版本,然后指定cudatoolkit版本来安装pytorch吧。例如系统的Nvidia Driver版本为440.33,查询到最高支持cudatoolkit版本为10.2,则可以使用conda install pytorch cudatoolkit=10.2 -c pytorch命令安装pytorch。

二、安装指令

安装cuda

这里cudatoolkit(cuda在conda中别名)指定你自己需要的版本

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

安装cudnn

直接安装cudnn,自动匹配你cuda合适的版本。

conda install cudnn

你可能感兴趣的:(深度学习,conda,pytorch,深度学习)