1. 下载和安装 Anaconda3
1.1 下载
下载页面:https://www.anaconda.com/download#downloads 下载地址:https://repo.anaconda.com/archive/Anaconda3-2023.03-1-Linux-x86_64.sh
下载成功后是一个 .sh
文件。
1.2 安装 Anaconda3
sh Anaconda3-2023.03-1-Linux-x86_64.sh
安装完成后,执行命令加载 conda
环境:. ~/.bashrc
2. 创建项目环境
2.1 创建项目环境 pytorchenv
conda create --name pytorchenv
2.2 使用指定的项目环境
activate pytorchenv
2.3 删除一个环境
conda remove -n your_env_name --all
2.4 删除包
conda remove --name $your_env_name $package_name
2.5 设置国内镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
设置搜索时显示通道地址
conda config --set show_channel_urls yes
2.6 打包conda虚拟环境
来源:https://blog.csdn.net/winycg/article/details/83759152
2.6.1 安装工具
pip install conda-pack
2.6.2 打包虚拟环境的目录
conda pack -n your_env_name -o your_env_name.tar.gz
若出现错误:CondaPackError: Cannot pack an environment with editable packages
可加上参数 --ignore-editable-packages
2.6.3 解压和使用
mkdir your_env_name
cd your_env_name
tar -zxf ../your_env_name.tar.gz
cd ..
source ./base_pytorch/bin/activate
2.7 直接打包和解压 conda 环境
tar zcvf your_env_name.tar.gz your_env_name
mkdir your_env_name
cd your_env_name
tar -zxf ../your_env_name.tar.gz
export PATH=/path/to/pth/bin:$PATH
2. 安装 PyTorch
安装命令获取地址:https://pytorch.org/get-started/locally/
2.1 安装 GPU 版本的 PyTorch
conda install pytorch torchvision torchaudio pytorch-cuda -c pytorch -c nvidia
2.2 安装 CPU 版本的 PyTorch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
备注
1. 还可以安装 LibTorch 版本的 PyTorch
来源:https://pytorch.org/get-started/locally/
Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-2.0.1%2Bcpu.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.0.1%2Bcpu.zip