安装Conda环境

下载并安装Anaconda或Miniconda,确保conda命令可用。推荐使用Miniconda以节省空间。安装完成后,打开终端或Anaconda Prompt进行后续操作。

创建新环境

使用conda创建一个新的虚拟环境,指定Python版本(PyTorch通常支持Python 3.7及以上版本):

conda create -n pytorch_env python=3.9

激活该环境:

conda activate pytorch_env

安装PyTorch

访问PyTorch官网(https://pytorch.org/),根据操作系统、硬件(CPU/GPU)和Conda/Pip偏好获取安装命令。例如,安装CPU版本的PyTorch:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

若使用GPU(需提前安装CUDA驱动),例如CUDA 11.3版本:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

验证安装

在Python交互环境中运行以下代码验证PyTorch是否正确安装:

import torch
print(torch.__version__)  # 输出PyTorch版本
print(torch.cuda.is_available())  # 检查GPU是否可用

管理依赖

导出当前环境的依赖列表以便复现:

conda env export > environment.yml

environment.yml恢复环境:

conda env create -f environment.yml

常见问题处理

若遇到冲突,可尝试清理缓存后重新安装:

conda clean --all
conda install --revision 0  # 回滚到初始状态

对于网络问题,可切换为国内镜像源(如清华源)加速下载:

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

Logo

火山引擎开发者社区是火山引擎打造的AI技术生态平台,聚焦Agent与大模型开发,提供豆包系列模型(图像/视频/视觉)、智能分析与会话工具,并配套评测集、动手实验室及行业案例库。社区通过技术沙龙、挑战赛等活动促进开发者成长,新用户可领50万Tokens权益,助力构建智能应用。

更多推荐