2019-10-04 pytorch 在win10下的安装

s天利用晚上这点时间,进行安装了pytorch。

pytorch的安装有两种:

1.conda

2.pip

我是用pip 安装的。

1.登录

 PyTorch(https://pytorch.org/)

2019-10-04 pytorch 在win10下的安装_第1张图片

点击get started

找到适合自己的版本,CUDA、平台、Python等,然后会自动提示“Run this command” 命令指令进行安装。我的是Python3.6.5(pycharm 2018.3.6 )+CUDA10.0+windows10,当然如果电脑没有GPU,CUDA就选none。

我的cuda是重新下载且安装的,去cuda官网下载。(所以没有经历查看NVIDA控制面板,我都没找到)

CUDA官网:CUDA Toolkit 10.1 Update 2 Download | NVIDIA Developer

2019-10-04 pytorch 在win10下的安装_第2张图片

2.按照要求选择

(1)准备工作:查看自己电脑的显卡版本(n卡)

①打开显卡控制面板

2019-10-04 pytorch 在win10下的安装_第3张图片

②选择帮助->系统信息

2019-10-04 pytorch 在win10下的安装_第4张图片

③选择组件,查看显卡版本,即为9.2.148,也即CUDA9

2019-10-04 pytorch 在win10下的安装_第5张图片

(2)按要求选择

我的是python 3.6+cuda 10+pip+windows

2019-10-04 pytorch 在win10下的安装_第6张图片

3、复制选项下面的两条命令到cmd安装

就是Run this Command:

pip3 install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

因国内下载很慢,我加了换源:-i https://pypi.tuna.tsinghua.edu.cn/simple 清华源

安装成功:

2019-10-04 pytorch 在win10下的安装_第7张图片

因为此处我的pip 是需要升级,所以升级了一下:

python -m pip install --upgrade pip

输入命令pip list,会看到安装列表:

2019-10-04 pytorch 在win10下的安装_第8张图片

4、测试一下

进入python,然后输入import torch,出现“>>>”且没报错那就算成功了

import torch导入没报错就说明安装成功,torch.cuda.is_available()是True就表示支持GPU啦~

2019-10-04 pytorch 在win10下的安装_第9张图片

torch.cuda.is_available()是True就表示支持GPU啦~

2019-10-04 pytorch 在win10下的安装_第10张图片

或者是在pycharm下:

2019-10-04 pytorch 在win10下的安装_第11张图片
2019-10-04 pytorch 在win10下的安装_第12张图片

你可能感兴趣的:(2019-10-04 pytorch 在win10下的安装)