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

📚博客主页:knighthood2001

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

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

前言

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

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

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

正文

首先,打开Anaconda Prompt

然后输入创建环境的命令

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

shell 复制代码
conda create -n pytorch python=3.8.2

中途可能需要输入y,表示继续安装。

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

shell 复制代码
conda env list

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

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

shell 复制代码
conda activate pytorch

可以看到,此时已经激活pytorch环境了。


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

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


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

shell 复制代码
nvidia-smi

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

建议使用pip进行下载。

shell 复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

这样就安装好了。

使用pip list查看一下库

shell 复制代码
pip list


注意,你看看这些版本后面有没有带cu的,如果没有,说明你下载的cpu版本的。

需要重新下载gpu版本的。

正确的如下:

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

pycharm设置

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

然后你就可以使用之前创建好的环境了。

相关推荐
心扬2 分钟前
python网络编程
开发语言·网络·python·tcp/ip
忧陌6065 分钟前
DAY 44 预训练模型
python
CareyWYR17 分钟前
每周AI论文速递(2506202-250606)
人工智能
YYXZZ。。21 分钟前
PyTorch——优化器(9)
pytorch·深度学习·计算机视觉
点云SLAM22 分钟前
PyTorch 中contiguous函数使用详解和代码演示
人工智能·pytorch·python·3d深度学习·contiguous函数·张量内存布局优化·张量操作
小天才才33 分钟前
【自然语言处理】大模型时代的数据标注(主动学习)
人工智能·机器学习·语言模型·自然语言处理
尘浮72835 分钟前
60天python训练计划----day45
开发语言·python
音程36 分钟前
预训练语言模型T5-11B的简要介绍
人工智能·语言模型·自然语言处理
哆啦A梦的口袋呀1 小时前
基于Python学习《Head First设计模式》第六章 命令模式
python·学习·设计模式
努力搬砖的咸鱼1 小时前
从零开始搭建 Pytest 测试框架(Python 3.8 + PyCharm 版)
python·pycharm·pytest