2019-10-04 pytorch 在win10下的安装
s天利用晚上这点时间,进行安装了pytorch。
pytorch的安装有两种:
1.conda
2.pip
我是用pip 安装的。
1.登录
PyTorch(https://pytorch.org/)
![](https://img.haomeiwen.com/i9099318/5d57aa4f7941c3d7.png)
点击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
![](https://img.haomeiwen.com/i9099318/541f0447b17a4031.png)
2.按照要求选择
(1)准备工作:查看自己电脑的显卡版本(n卡)
①打开显卡控制面板
![](https://img.haomeiwen.com/i9099318/9f928b9e94ffd75f.png)
②选择帮助->系统信息
![](https://img.haomeiwen.com/i9099318/c1499a8d3de37493.png)
③选择组件,查看显卡版本,即为9.2.148,也即CUDA9
![](https://img.haomeiwen.com/i9099318/277379352c8af430.png)
(2)按要求选择
我的是python 3.6+cuda 10+pip+windows
![](https://img.haomeiwen.com/i9099318/8324937901ed5d86.png)
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 清华源
安装成功:
![](https://img.haomeiwen.com/i9099318/d50835c21519990d.png)
因为此处我的pip 是需要升级,所以升级了一下:
python -m pip install --upgrade pip
输入命令pip list,会看到安装列表:
![](https://img.haomeiwen.com/i9099318/e010a73773e99ee6.png)
4、测试一下
进入python,然后输入import torch,出现“>>>”且没报错那就算成功了
import torch导入没报错就说明安装成功,torch.cuda.is_available()是True就表示支持GPU啦~
![](https://img.haomeiwen.com/i9099318/ed40310965cae865.png)
torch.cuda.is_available()是True就表示支持GPU啦~
![](https://img.haomeiwen.com/i9099318/221566e961ee645a.png)
或者是在pycharm下:
![](https://img.haomeiwen.com/i9099318/b18d219b6d8c59d7.png)
![](https://img.haomeiwen.com/i9099318/339f8af8bde0a41a.png)