首先查看自己安装的是否是GPU版本
一开始用的版本比较高,但是发现是CPU的,所以不要安装
我们去选择比较低版本的pytorch
就能看到是GPU版本的
选择yes,遇到比较大的文件包,一般都会下载不成功,所以要多下载几次。首先遇到了下面这个问题,原因是网速不好,可以重新下几次或者使用清华源下载。
下完后又出现了下面的问题
一开始不敢多删。后面发现只要按它的提示去把对应目录下的cudnn64_7.dll文件删除
后面回到虚拟环境继续打安装指令(我是打了后才去看安装是否成功的,应该是不需要再打安装指令,直接查看虚拟环境下的安装包)
然后验证安装是否成功,输入python
得到true就代表安装成功了。