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

一、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。

猜你喜欢

转载自blog.csdn.net/wzk4869/article/details/126737684#comments_25997861
今日推荐