系统是Windows,由于使用pytorch官网上推荐的语句下载pytorch总会导致下载的版本为cpu版本,在经过大量尝试后,发现可以使用如下方法手动下载gpu版本:
1、更改系统变量里的python路径为想要安装pytorch的那个conda虚拟环境的Python路径,注意用户变量和系统变量里面都做修改。
2、在如下网址中找到所需pytorch版本对应的文件:
https://download.pytorch.org/whl/torch/
3、下载上述网址中的对应的.whl文件到一个指定的文件夹,然后用cmd进入此文件夹,使用如下语句:
python -m pip install name.whl
其中naem.whl就是下载的文件。