Ubuntu下PyTorch开发环境配置

文章目录

• 前言
• 安装Anaconda
• 配置Python环境
• 安装Pytorch
• 开发IDE
• 最后

前言

你好，我是醉墨居士，今天在Ubuntu上配置一下PyTorch开发环境，同时记录一下我的经历，希望能够让你快速完成开发环境的配置，节约宝贵的时间

安装Anaconda

• 安装脚本下载

wget https://repo.anaconda.com/archive/Anaconda3-2024.06-1-Linux-x86_64.sh

交互说明：使用默认选项一路回车即可

• 配置anaconda环境变量

conda init bash
source ~/.bashrc

配置Python环境

• 创建python环境

conda create -n zm python=3.12

交互说明：使用默认选项回车即可

• 切换到zm环境

conda activate zm

安装Pytorch

PyTorch官网这里提供相关硬件条件下的不同安装命令，大家根据自身情况，执行适合自己的安装命令即可
https://pytorch.org/get-started/locally/

我这里以cpu版本的安装为例

• cpu版本

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

开发IDE

我这里推荐使用PyCharm，使用体验良好

其它可选的开发编辑器还有Visual Studio Code，Jupyter Notebook

最后

至此，少年恭喜你，已经完成了开发环境的配置，希望你能在未来深度学习的征程中一帆风顺