手把手教你用Anaconda配置PyTorch环境(避坑指南)

前言:为什么需要虚拟环境?

搞机器学习的朋友都知道(特别是刚入门的新手),不同项目需要的库版本可能互相冲突!比如这个项目要用PyTorch 1.8,那个项目又要用PyTorch 2.0,这时候虚拟环境就是你的救命稻草(别问我怎么知道的,都是血泪教训😭)

第一步:安装Anaconda全家桶

  1. 官方下载清华镜像站比官网快10倍(你懂的)
  2. 安装过程:记住一定要勾选"Add to PATH"(新手不勾这个后续会哭晕在厕所)
  3. 验证安装:打开CMD输入conda -V,看到版本号才算成功(别手抖输成conda-v啊!)

创建专属训练场(虚拟环境)

conda create -n pytorch_env python=3.8  # 建议用Python3.8(兼容性之王)
conda activate pytorch_env              # 进入你的专属空间

注意!看到命令行前面出现(pytorch_env)才说明激活成功(很多新手栽在这里)

重头戏:PyTorch安装的玄学操作

官方安装命令(容易翻车版)

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

这里有个大坑!-c pytorch参数会强制从官方源下载(速度堪比龟速),建议先配置清华镜像源

镜像加速方案(建议收藏)

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

终极安装命令(带cuda版)

conda install pytorch torchvision torchaudio cudatoolkit=11.3

这里有个隐藏技巧:先运行nvidia-smi查看你的CUDA版本(N卡用户必备),如果显示"CUDA Version: 11.6"就把上面的11.3改成11.6

验证安装是否成功

import torch
print(torch.__version__)          # 输出版本号就算成功?
print(torch.cuda.is_available())  # 显示True才是真的成功!

如果第二个输出是False,可能是:

  1. 没装NVIDIA驱动(去官网下!)
  2. CUDA版本不匹配(重装对应版本)
  3. 安装时漏了cudatoolkit参数

常见问题Q&A

Q:安装时提示PackagesNotFoundError怎么办?
A:八成是拼写错误!检查是不是把pytorch写成pytroch了(亲身经历警告)

Q:CPU版本和GPU版本怎么选?
A:有N卡就选GPU版(速度飞起),笔记本用户建议用CPU版(避免烧主板)

Q:为什么我的显存显示0MB?
A:八成是驱动问题!更新驱动后重启,还不行就重装CUDA

终极技巧:环境迁移大法

把当前环境打包:

conda env export > environment.yaml

在新电脑上复现:

conda env create -f environment.yaml

结语

配置环境就像搭积木,虽然过程可能让人抓狂(特别是网络不好的时候),但只要按照步骤来,99%的问题都能解决。记住:所有报错都是纸老虎!遇到问题先去喝杯咖啡,回来再战~

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值