PyTorch安装及配置（详细的步骤）

浅浅的花香味﹌ 2023-09-27 08:47 145阅读 0赞

### 一、Anaconda安装及配置 ###

这一步我前面的博客已经写过，这里就不再赘述！

> https://blog.csdn.net/wzk4869/article/details/126714956?spm=1001.2014.3001.5502

### 二、PyTorch的下载及配置 ###

**1.创建虚拟环境**

在下载PyTorch之前，首先我们需要创建一个虚拟环境，在电脑左下角的“开始”栏中找到`Anaconda Prompt`，点击打开：

![5cf1cb003b88422d8cc172cfd677040b.png][]

输入以下代码`conda create -n PyTorch python=3.7`

这里解释一下这行代码，表示创建一个名叫PyTorch的环境，其中的Python解释器版本为3.7，这个需要根据大家下载的python版本写。

输入代码回车后，会出现`([y]/[n]?)`，输入`y`等待安装完毕后，我们的虚拟环境便创建好了。

**2.在虚拟环境中下载PyTorch**

创建完虚拟环境后，打开Anaconda Prompt，输入`conda activate PyTorch`，激活所创建的虚拟环境。

然后进入到pytorch的官网：

> https://pytorch.org/get-started/locally/

![0d588539045044eea4cbec0ca932f07d.png][]  
电脑没有显卡的选择`CPU版本`，有显卡的可以选择`CUDA版本`的（CPU版本的在后续大量数据训练中速度肯定会更慢一点）。

打开cmd控制台，输入`nvidia-smi`查看本机的cuda版本:

![18dac9f820074c41b0366475302b2453.png][]  
比如我的CUDA版本是11.6，可以向下兼容，可以选择10.2、11.3版本的PyTorch。

因为10.2目前停止下载，所以我选择11.3的版本：

![在这里插入图片描述][a918602b161545c299db34a12cf6607a.png]

复制我们这段代码放入刚刚创建的虚拟环境中：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

会出现`([y]/[n]?)`，输入`y`，然后等待它下载完成即可。

下载完成后，我们验证一下是否成功下载好了PyTorch。

我们打开Pycharm，配置刚刚建好的虚拟环境：

import torch
    a=torch.cuda.is_available()
    print(a)

若结果为`True`，则表示配置成功！

![d8ca19818f39478b9595bd2578510657.png][]

### 三、本机CUDA版本低怎么下载CUDA高版本的Pytorch ###

我的电脑最开始的cuda版是较低的，而不是上面显示的11.6，这是怎么一回事呢？

几乎所有的笔记本电脑所支持的cuda版本是可以升级的，我的显卡是英伟达的，打开NVIDIA控制面板查看显卡类型，我的是GeForce 940MX：

![在这里插入图片描述][36189c4726694edfb9d73d54acdf3bdf.png]

然后登陆英伟达的官网：

> https://www.nvidia.cn/Download/index.aspx?lang=cn

![8c7d49d745684ea080c21fc5e7122efc.png][]  
选择与自己电脑显卡型号相同的选项（带Notebooks的是笔记本），然后点击搜索：

最后点击下载即可。后面的安装教程也比较简单，这里就不再讲诉了，大家可以看看这篇博客：

> https://blog.csdn.net/m0\_37870649/article/details/105356540

安装完成后，再次查看电脑的cuda版本可以发现已经升级到最新，此时可以按照之前的方法下载CUDA为11.3的Pytorch。

[5cf1cb003b88422d8cc172cfd677040b.png]: https://img-blog.csdnimg.cn/5cf1cb003b88422d8cc172cfd677040b.png
[0d588539045044eea4cbec0ca932f07d.png]: https://img-blog.csdnimg.cn/0d588539045044eea4cbec0ca932f07d.png
[18dac9f820074c41b0366475302b2453.png]: https://img-blog.csdnimg.cn/18dac9f820074c41b0366475302b2453.png
[a918602b161545c299db34a12cf6607a.png]: https://img-blog.csdnimg.cn/a918602b161545c299db34a12cf6607a.png
[d8ca19818f39478b9595bd2578510657.png]: https://img-blog.csdnimg.cn/d8ca19818f39478b9595bd2578510657.png
[36189c4726694edfb9d73d54acdf3bdf.png]: https://img-blog.csdnimg.cn/36189c4726694edfb9d73d54acdf3bdf.png
[8c7d49d745684ea080c21fc5e7122efc.png]: https://img-blog.csdnimg.cn/8c7d49d745684ea080c21fc5e7122efc.png