anaconda创建并且配置pytorch(完整版)

发布于:2024-08-11 ⋅ 阅读:(30) ⋅ 点赞:(0)

📚博客主页:knighthood2001
公众号:认知up吧 **
🎃
知识星球:【认知up吧|成长|副业】介绍**
❤️如遇文章付费,可先看看我公众号中是否发布免费文章❤️
🙏笔者水平有限,欢迎各位大佬指点,相互学习进步!

当你深度学习玩多了,可能就体会到conda的方便了。–认知up吧

前言

这篇文章改了一次,因为我之前使用清华源的原因,导致pytorch下载下来是个cpu版本的,深度学习要是没有gpu,你还玩个屁,电脑都撸冒烟了要。

如果你也下载成cpu版本,你可以通过路径,找到具体的pytorch库,然后把相关文件夹都删掉。

我这下面的已经是gpu版本了。

在这里插入图片描述

正文

首先,打开Anaconda Prompt
在这里插入图片描述

然后输入创建环境的命令

我这里创建一个pytorch的环境名称,使用3.8.2的python版本。因此,输入命令行如下

conda create -n pytorch python=3.8.2

在这里插入图片描述
中途可能需要输入y,表示继续安装。

然后你可以使用如下命令,去查看你配置的所有环境

conda env list

在这里插入图片描述
可以看到,环境已经有了,我这里的环境都是放到了c盘以外的,如果需要解决环境迁移的,可以看我之前写的这篇文章,因为pytorch这种框架,动辄3个G,放在c盘,可能占用较大内存。
anaconda修改安装的默认环境

接着,你需要激活一下环境,后续才能在这里面安装pytorch框架,否则,关着门,是装不了东西的。

conda activate pytorch

在这里插入图片描述
可以看到,此时已经激活pytorch环境了。


首先,pytorch库挺大的,下载慢特别正常。

最重点的是,千万不要去设置清华源,如果使用清华源下载的话,下载下来的pytorch是cpu版本的,后续你有显卡,也发挥不出作用。


然后去pytorch官网上找对应命令,这里大家最好在cmd中输入nvidia-smi查看一下你的cuda版本。

nvidia-smi

在这里插入图片描述

官网地址:https://pytorch.org/get-started/locally/#windows-pip

在这里插入图片描述
建议使用pip进行下载。

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

在这里插入图片描述

这样就安装好了。

使用pip list查看一下库

pip list

在这里插入图片描述
注意,你看看这些版本后面有没有带cu的,如果没有,说明你下载的cpu版本的。

需要重新下载gpu版本的。

正确的如下:

在这里插入图片描述

以后你就可以去用这个环境,进行深度学习的环境配置了。

pycharm设置

打开pycharm,进行如下设置,注意conda可执行文件,一般是在你下载anaconda下的scripts下的conda.exe

在这里插入图片描述

在这里插入图片描述
然后你就可以使用之前创建好的环境了。