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
最后
至此，少年恭喜你，已经完成了开发环境的配置，希望你能在未来深度学习的征程中一帆风顺