开始的anaconda安装,cuda安装和cudnn安装的步骤就不再说了,比较简单,一般不会出现错误,主要在pytorch安装上,许多人明明安装好了,但是在验证时,却显示为False,这里说一个无脑入的方法,让你轻松得到True
下载地址,进入下载pytorch的地址,如果下载cuda版本,记得一直往下翻,一直翻到,前面对应有cu……的版本,前面没cu的为cpu版本,如图
比如cu102,就是对应的cuda版本为10.2的然后为torch,torchvision,torchaudio,后面为它的版本号,cp37为python37用的
比如我的cuda安装的是11.1的,然后python版本为3.8的,在Windows下的,所以我下载了
torch和torchvision(注意torch和torchvision的版本号一定要对应好,文章下面有对应的表格,其次我们先下载torch,再下载torchvision)
首先进入我们的环境,然后输入命令pip install +"下载的文件地址“
提醒!!!!!!:
1,一定要对应好你的torch和torchvision的版本号,不然在你下载torchvision的时候,可能会卸载你原来的torch,会导致最终结果为false
2.先安装torch,torch vision
验证是否安装成功了
我们输入
python
import torch
print(torch.cuda.is_available())
当结果为TRUE时,即安装成功!!!!!


