首先通过 python 官网将python安装,这一步先跳过了

直接进入主题 torch 安装

https://pytorch.org/

首先要进入 pytorch 官网(以给出)进入如下界面

pytorch速查表 pytorchapi_官网

 

点击 install 进入到配置选择的界面

 

pytorch速查表 pytorchapi_pytorch速查表_02

  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 来构建自己的深度学习模型了

pytorch速查表 pytorchapi_html_03

但是 往往安装起来并不是那么简单

最常见的错误就是“找不到模块”“找不到安装包”等错误,而一般情况下可以通过以下方法解决,因为存在 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

 

愿我们进步的路上不孤单