Pytorch安装

一.Anconda的环境配置

1. 使用清华镜像

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --remove channels defaults
conda config --set show_channel_urls yes

查看镜像

conda config --show channels

移除镜像

conda config --remove-key channels

2. 创建环境pytorch，使用Python版本是3.7

也可以使用原环境跳过这个步骤

conda create -n pytorch python=3.7 
conda activate pytorch 

3. 查看环境是否安装成功

conda info --envs

4. 安装pytorch

1. 网站：https://pytorch.org/

2. 查看CUDA版本

nvcc --version

3. 运行代码

mamba install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

# conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

# conda install torch===1.7.1 torchvision===0.8.2 torchaudio===0.7.2 cudatoolkit=11.1 -c pytorch -c conda-forge

5. 切换conda环境及测试

1.查看已有的虚拟环境

conda env list

2.切换到想要的虚拟环境，这里我切换到pytorch

conda activate pytorch

3.测试

python
import torch
print(torch.cuda.is_available()) # 返回true表示pytorch安装完成