PyTorch安装及配置(详细的步骤)

  • Post author:
  • Post category:其他




一、Anaconda安装及配置

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

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



二、PyTorch的下载及配置


1.创建虚拟环境

在下载PyTorch之前,首先我们需要创建一个虚拟环境,在电脑左下角的“开始”栏中找到

Anaconda Prompt

,点击打开:

输入以下代码

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/



电脑没有显卡的选择

CPU版本

,有显卡的可以选择

CUDA版本

的(CPU版本的在后续大量数据训练中速度肯定会更慢一点)。

打开cmd控制台,输入

nvidia-smi

查看本机的cuda版本:



比如我的CUDA版本是11.6,可以向下兼容,可以选择10.2、11.3版本的PyTorch。

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

在这里插入图片描述

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

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

,则表示配置成功!



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

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

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

在这里插入图片描述

然后登陆英伟达的官网:

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



选择与自己电脑显卡型号相同的选项(带Notebooks的是笔记本),然后点击搜索:

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

https://blog.csdn.net/m0_37870649/article/details/105356540

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



版权声明:本文为wzk4869原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。