安装pytorch(个人经验总结,有效)

  • 1 安装好anaconda 、pycharm和python本体
  • 2 提前下载好torch的两个whl文件
  • 3 在pycharm中创建新工程并创建一个.py文件
  • 4 安装两个whl文件
  • 5 把安装好的环境链接到工程上
  • 6 验证是否安装成功


1 安装好anaconda 、pycharm和python本体

anaconda 是一个管家,相当于碗(环境);pycharm是一个IDE,相当于筷子,以及最重要的python本体,这是锅里的饭,但是没人就着锅吃饭,需要筷子和碗配合。

2 提前下载好torch的两个whl文件

①网址:文件下载链接: link.

②查找对应版号:如图所示,图中版本为1.7.1

pytorch:link. 这个网址是提供安装pytorch指令的。

pycharm域pytorch区别 pytorch和pycharm的区别_python

选择对应你系统版本的指令,并查看你下载的pytorch版本,在①中的网址下载对应的whl文件。
torchvision:对应pytorch版本下载对应的whl文件。下载链接和pytorch后的链接一样。
torchvision与pytorch对应关系:link.

3 在pycharm中创建新工程并创建一个.py文件

在pycharm中创建新工程并创建一个.py文件,在该页面下方的“terminal”中创建一个新的虚拟环境:conda create -n mypytorch01 python=3.8
然后激活环境:

# To activate this environment, use
#
#     $ conda activate mypytorch01  
#
# To deactivate an active environment, use
#
#     $ conda deactivate

4 安装两个whl文件

①把两个文件复制到你创建的工程所在文件夹,否则会报错:“WARNING: Requirement ‘torch-1.7.1+cpu-cp38-cp38-win_amd64.whl’ looks like a filename, but the file does not exist”
②使用一般的指令下载网速极其慢,使用下面这个大神在豆瓣(没错,就是你吃瓜的那个豆瓣)搞的指令:

pip --default-timeout=100 install 库名称 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

其中“库名称”需要改为你的whl文件的文件名,注意需要用英文的双引号。
示例:
pytorch:

pip --default-timeout=100 install "torch-1.7.1+cpu-cp38-cp38-win_amd64.whl" -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

torchvision:

pip --default-timeout=100 install "torchvision-0.8.2+cpu-cp38-cp38-win_amd64.whl" -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

这是我使用的对应我的电脑版本的torch与torchvision的安装指令,只需要把双引号里的库名称改为你自己的即可。

5 把安装好的环境链接到工程上

打开pycharm,file—settings—python interpreter—齿轮按钮—Add—conda environment—existing environment—三个点(…)—点击小眼睛显示隐藏文件—找到你安装的anaconda—envs—你创建的工程文件—python.exe—一路下来点击OK。

最终就能看到你需要的一切全部安装好了。

pycharm域pytorch区别 pytorch和pycharm的区别_python_02

6 验证是否安装成功

在pycharm主页面输入:

import torch
print("hello pytotorch.__version__rch{}".format(torch.__version__))

证明torch安装成功:

pycharm域pytorch区别 pytorch和pycharm的区别_anaconda_03


参考视频:哔哩哔哩干杯.