【Deepin 20系统】Linux系统Anaconda安装Pytorch

向右看齐 2022-10-31 00:59 369阅读 0赞

1 创建环境

1.1 安装Anaconda

参考本人另一篇博客【Deepin 20系统】Linux系统安装Anaconda和Tensorflow-gpu2.3

1.2 创建一个conda的环境

本人选择的python 版本3.6

  1. conda create -n Pytorch python=3.6
  2. # 环境名称：Pytorch python版本：3.6

进入环境

  1. conda activate Pytorch

2 查看系统Cuda版本

  1. cat /usr/local/cuda/version.txt

3 找到对应版本的执行命令

进入官网选择对应cuda版本的命令。本人的cuda是11的。
在这里插入图片描述

  1. conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

但是别着急执行命令，因为pytorch有700多M，下载很慢，需要添加镜像后使用

4 添加镜像源

参考conda安装Pytorch下载过慢解决办法(11月26日更新ubuntu下pytorch1.3安装方法)

  1. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  2. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  3. conda config --set show_channel_urls yes
  5. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  6. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  7. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  8. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  9. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

5 执行安装命令

注意：要想通过镜像下载，去掉官网提供的命令的-c pytorch

  1. conda install pytorch torchvision torchaudio cudatoolkit=11.0

发表评论

表情：
评论列表 （有 0 条评论，369人围观）

还没有评论，来说两句吧...

相关阅读