Linux虚拟环境安装cuda、cudnn、pytorch
conda虚拟环境中安装cuda和cudnn
·
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())
更多推荐
已为社区贡献1条内容
所有评论(0)