手把手教你用Anaconda搭建PyTorch深度学习环境(避坑指南)

最近好多小伙伴私信问我PyTorch环境配置总出错怎么办?今天我就用最接地气的方式,带大家用Anaconda三步搞定PyTorch环境搭建!(文末有彩蛋)


🚀 准备工作(别急着跳步骤!)

  1. 安装Anaconda
    清华镜像站下载最新版(2023年推荐选Python3.9版本)。安装时千万记得勾选👇
    ✅ Add Anaconda to my PATH environment variable
    (不然之后命令行都用不了,别问我怎么知道的🙃)

  2. 换国内镜像源(速度飙升10倍!)
    打开终端输入:

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

💡 核心操作(重点来了!)

步骤1:创建独立环境

conda create -n pytorch_env python=3.8

(为什么用3.8?因为最新版PyTorch可能还没适配最新Python版本,这里求稳)

步骤2:激活环境

conda activate pytorch_env

看到命令提示符前面出现(pytorch_env)才算成功!(新手常在这里翻车)

步骤3:安装PyTorch

打开PyTorch官网选择你的配置:

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

(重点提醒!CUDA版本要和显卡驱动匹配,不确定的看文末常见问题)


🚨 常见坑点(血泪经验!)

  1. CUDA版本不对应
    运行nvidia-smi查看支持的CUDA最高版本,比如显示11.7就选cudatoolkit=11.7

  2. conda和pip混用
    要么全用conda,要么全用pip!混用会导致依赖冲突(别问我怎么解决,重装更快🙈)

  3. 镜像源没生效
    如果下载速度慢,试试:

    conda clean -i
    conda config --remove-key channels
    

    重新配置镜像源


✅ 验证安装(必须做!)

打开Python解释器输入:

import torch
print(torch.__version__)  # 查看版本
print(torch.cuda.is_available())  # 显示True才算GPU可用!

🌈 终极技巧(官网不会告诉你)

  1. 离线安装法
    清华镜像下载对应版本的.tar.bz2文件,然后:

    conda install /path/to/package.tar.bz2
    
  2. 环境克隆大法
    做好环境后记得:

    conda env export > environment.yaml
    

    下次直接conda env create -f environment.yaml秒恢复!


📝 最后说两句

配置环境就像搭乐高,按步骤来其实很简单!建议先创建虚拟环境再安装,这样玩崩了也不怕(大不了删了重来)。遇到报错别慌,把错误信息复制到Google,90%的问题都有现成解决方案!

(悄悄说:最近发现用PyCharm直接管理conda环境更方便,想看的评论区扣1,下期安排!)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值