Windows 10 安装PyTorch
安装过程
今天比较有时间,正好看完了CVPR-2018年的论文<**Single Image Reflection Separation with Perceptual Losses**>,这里不多介绍这篇论文相关的东西了。因为跑这个数据的话,自己的电脑当时在Windows下装的Tensorflow,但是作者给的运行策略是Ubuntu下,所以问了一下学长,他说可以在Windows下安装PyTorch来跑。下面就记录一下自己安装PyTorch的过程。
1.由于之前安装了Anaconda3,所以可以采用conda安装的方式,但是之前安装Tensorflow的时候Python版本的问题就一直有点疑问,所以查了一下发现如果只是想利用Pycharm运行PyTorch的话,其实直接用pip方式安装就可以
2.那么需要到PyTorch官网去查看自己所需的代码 PyTorch官网连接,相应的选择自己电脑的配置,注意如果用pip安装的话,要选择Pip选项。然后下面就会给出你所需的命令。还要注意的一点是我的电脑由于是A卡的,所以在CUDA这个部分我选择的是None,也就是所说的CPU版本的PyTorch。本人的命令如下:
pip3 install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
3.安装成功后,一定要记得看一下自己所安装的文件夹地址
4.打开Pycharm,按ctrl+alt+s,打开Project Interpreter进行配置,点击右上角的设置符号,选择Add,在下面点选Existing environment,然后在右边的…处找到自己刚才安装的文件夹地址,拉到最下面找到python.exe文件一路点击ok。
5.在pycharm中新建项目,输入以下代码:
from __future__ import print_function
import torch
x=torch.rand(5,3)
print(x)
查看输出,如果输出为:
tensor([[0.9072, 0.3848, 0.9203],
[0.1643, 0.7447, 0.1730],
[0.4379, 0.2974, 0.1283],
[0.4563, 0.5322, 0.9924],
[0.4715, 0.9287, 0.7176]])
Process finished with exit code 0
证明PyTorch安装成功!