通过 pip 安装 pytorch

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

直接进入主题 torch 安装

https://pytorch.org/

首先要进入 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

 





posted @ 2020-11-14 12:30  洋聪love  阅读(37212)  评论(2编辑  收藏  举报