之前安装过GPU版本的pytorch,但是不知道什么原因虚拟环境无法使用,重新安装是注意要查看GUDA是否存在
官方验证步骤:/cuda/cuda-installation-guide-microsoft-windows/index.html#compiling-examples__valid-results-from-sample-cuda-bandwidthtest-program
在此之前,可以添加环境变量。这里我按照官网上的步骤并没有添加
First:通过nvcc -V验证
1、按Windows+R,输入cmd,进去cmd.exe
2、输入nvcc -V
3、显示类型下图的版本信息
在这里插入图片描述
还有另外的方法进行验证
Second: 运行deviceQuery或bandwidthTest 示例程序
然后重新下载相应版本的cuDNN
最后重建 虚拟环境 安装pytorch