Step1:打开pytorch官网

进入pytorch官网,向下拉,可以看到如下界面
深度学习之PyTorch安装
此时完全是默认的选项,第一个选择版本,建议选择稳定版。
第二个是自己的操作系统,我的是windows系统。
第三个是安装的来源,喜欢那个就选哪个,此处我们用默认的conda,你也可以选择pip
第四个是语言,默认是python
第五个是电脑所支持的CUDA版本,如果不支持可以选择NONE。

简单介绍一下如何查看自己的电脑是否支持CUDA

一般集成显卡是不支持CUDA,本人显卡是NVIDIA,以此为例来解释一下如何判断是否支持CUDA,以及支持哪个版本。

1、首先打开NVIDIA控制面板

深度学习之PyTorch安装

2、点击帮助->系统信息->组件

深度学习之PyTorch安装
这样就可以看到自己的系统所支持的CUDA的版本啦。

step2 安装pytorch(使用PIP安装是可以加速的哦)

在pytorch官网选择好相应的选项之后,将代码直接复制粘贴到anaconda的prompt之后运行即可。(如果是选择的pip,可以直接粘贴到cmd运行)
如果你使用PIP安装的话,想要加快安装速度,可以参考我上一篇文章的介绍:解决pip install时下载速度慢的问题–使用镜像
深度学习之PyTorch安装
深度学习之PyTorch安装
接下来便是漫长的等待工作啦。

step3 检查PyTorch是否安装成功

打开jupyter notebook,输入以下命令
深度学习之PyTorch安装
如果和图片显示一致,则安装成功!

相关文章: