通过 pip 安装 pytorch
首先通过 python 官网将python安装,这一步先跳过了
直接进入主题 torch 安装
首先要进入 pytorch 官网(以给出)进入如下界面
点击 install 进入到配置选择的界面
Pytorch Build 这一行可供选择的是 稳定版(左侧)和 试用版 (右侧)
your OS 这里选择属于自己的电脑系统
Package 这里需要选择的是通过怎样的途径下载,本文以 pip 为例子进行安装操作
CUDA 在自己电脑上的系统信息中查看电脑的CUDA型号,这里实际上是确认电脑是否支持 GPU 计算
在选择好这些之后,在 Run this command 中会给出下载链接,将
pip install torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
复制之后,win+R 输入 cmd 打开,将复制内容粘贴到这里就可以开始下载安装了
一般情况到这里就基本结束了,可以在命令行中进去python环境进行检测,这样说明安装成功,就可以利用 pytorch 来构建自己的深度学习模型了
但是 往往安装起来并不是那么简单
最常见的错误就是“找不到模块”“找不到安装包”等错误,而一般情况下可以通过以下方法解决,因为存在 python 和 pytorch 模块兼容问题,所以会存在python版本为3.7只能安装torch1.5 和torchvision0.6,
所以如果你安装的python版本是3.7或者之前的版本,那么在下载安装 pytorch 时就要考虑下载合适的版本
法1:可以在 pytorch 官网复制那个下载链接是将
pip install torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
改为
pip install torch===1.5.0 torchvision===0.6.1 torchaudio===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html
基本上就可以解决问题了
法2:通过如下链接下载对应的torch安装包来直接下载安装
https://download.pytorch.org/whl/torch_stable.html
更新1
问题:在调用 torch.from_numpy 时会出现 Module 'torch' has no 'from_numpy' memberpylint(no-member)的错误
解决:通过快捷键 Ctrl+Shift+P
打开vscode的命令面板,输入settings.json
并打开第一项,将如下内容复制进去并保存,即可解决问题
"python.linting.pylintArgs": [
"--errors-only",
"--generated-members=numpy.*, torch.*, cv2.*, cv.*"
]
更新2
问题:vs code 运行 torch.rand() 时出现 “Module 'torch' has no 'rand' memberpylint(no-member)” 的错误
解决:通过快捷键 Ctrl+Shift+P
打开vscode的命令面板,输入settings.json
并打开第一项,将如下内容复制进去并保存,即可解决问题
// whitelist torch to remove lint errors
"python.linting.pylintArgs": [
"--extension-pkg-whitelist=torch",
"--ignored-modules=torch",
"--ignored-classes=torch",
]
pip 清华园安装
-i https://pypi.tuna.tsinghua.edu.cn/simple