使用 conda 安装 pytorch

创建日期: 2023-06-20 16:46 | 作者: 风波 | 浏览次数: 17 | 分类: PyTorch

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
17 浏览
10 爬虫
0 评论