Linux虚拟环境安装cuda、cudnn、pytorch

来自http://t.csdn.cn/WSvvX

1.创建虚拟环境

conda create -n environment_name

激活虚拟环境：

conda cactivate environment_name

2.安装cuda

首先查出源内所有可用cuda版本：

conda search cudatoolkit --info

复制你想下载的版本对应的url（如图所示的cuda版本就是11.7）

 

将cuda包下载到本地，你想下载到哪个文件夹，就在该文件夹下右键运行终端

wget 刚复制的url链接

安装cuda：

conda install --use-local 本地cuda包所在的路径

3.安装cudnn

查看cuda对应的cudnn版本：

conda search cudnn --info

选择对应版本

-cudatoolkit 11.*即说明适合所有11开头的cuda版本

 

 选择之后接下来步骤和下载cuda一样

4.安装pytorch

查找和cuda匹配的pytorch版本pytorch官网

可以在previous pytorch versions里面看到之前的版本

注意：直接复制官网上的命令，不要把-c python去掉，否则下载的很有可能还是cpu版本的

 5.验证

验证cuda

print(torch.version.cuda)

验证cudnn

print(torch.backends.cudnn.version())