来自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())

 

Logo

更多推荐