- 安装好anaconda
- 查看自己电脑的cuda版本
nvidia-smi
然后根据cuda版本找合适的pytorch和torchvision版本:下面链接查询官网
Previous PyTorch Versions | PyTorch https://pytorch.org/get-started/previous-versions/
这里不一定是这个版本,只要能与cuda版本对应的都可以选,我选择了torch 1.10.1版本,然后我的python版本是3.6,windows 64位的系统
去下面链接下载相应的.whl文件
https://download.pytorch.org/whl/torch_stable.html
将该文件下载到你的user路径下面,
注意这里不是下载到你的虚拟环境路径下。
下载完毕后进入你的虚拟环境,安装
pip install torch-1.10.1+cu111-cp36-cp36m-win_amd64.whl
若出现错误:
只需要更新你的pip即可解决
命令:python -m pip install --upgrade pip
更新成功后重新安装
若报错:ERROR: Package 'torch' requires a different Python: 3.6.0 not in '>=3.6.2'
将环境中python版本替换位所需要版本即可:
pip install python==3.6.2 -n “你的虚拟环境名称”
最后再重新输入安装命令即可。
安装成功后,输入python,然后输入import torch
输出True则表示安装成功!!!
安装成功之后可以将.whl移到其他路径或者删除(如果你保存那个盘的内存不够的情况下考虑,若内存足够可忽略)
参考:https://blog.csdn.net/pxyp123/article/details/123747978