文章目录
前言:为什么需要虚拟环境?
搞机器学习的朋友都知道(特别是刚入门的新手),不同项目需要的库版本可能互相冲突!比如这个项目要用PyTorch 1.8,那个项目又要用PyTorch 2.0,这时候虚拟环境就是你的救命稻草(别问我怎么知道的,都是血泪教训😭)
第一步:安装Anaconda全家桶
- 官方下载:清华镜像站比官网快10倍(你懂的)
- 安装过程:记住一定要勾选"Add to PATH"(新手不勾这个后续会哭晕在厕所)
- 验证安装:打开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,可能是:
- 没装NVIDIA驱动(去官网下!)
- CUDA版本不匹配(重装对应版本)
- 安装时漏了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%的问题都能解决。记住:所有报错都是纸老虎!遇到问题先去喝杯咖啡,回来再战~