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安装完成

__EOF__

本文作者：skypanxh
本文链接：https://www.cnblogs.com/skypanxh/p/15214553.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！