安装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指令的。
选择对应你系统版本的指令,并查看你下载的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。
最终就能看到你需要的一切全部安装好了。
6 验证是否安装成功
在pycharm主页面输入:
import torch
print("hello pytotorch.__version__rch{}".format(torch.__version__))
证明torch安装成功:
参考视频:哔哩哔哩干杯.