anaconda安装gpu版pytorch(conda+pip+离线方法)

anaconda安装gpu版pytorch(cuda11.1)

众所周知，在anaconda安装库什么的属实有点玄学，有时一下就安好了，有时候一直报不知名错误，所以，特地将安装gpu版pytorch的方法总结一下，三种方法，总有一种能行！

–默认已安好anaconda，打开anaconda prompt，以及创建好虚拟环境并进入✌

activate env_name

conda安装

打开pytorch官方网站，点击Get Started.

1. 选择相关配置及要安装的cuda版本。

2. 输入Run this Command 中的运行指令安装。
   -c pytorch：说明用国外网站下载，速度较慢。可改用国内镜像网站下载，此时需先添加源，直接复制输入：

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

   添加后，可使用下指令进行查看：

    conda config --show-sources
   

   运行安装指令，此时仅输入-c pytorch前的指令。

tip：若无想要的cuda版本，可采取以下两种方法：(官网一般显示最新版本)
（1）更改指令中cuda版本号为自定义版本（11.1）（安装cuda11.1版本须在指令最后添加 (空格)-c conda-forge）
（2）离线安装

pip安装

操作同conda安装，注意将Package更改为Pip。
Run this Command 运行指令中的各库版本号可自行更改

离线安装

1 在网站上下载对应配置的pytorch的安装文件至本地。(二选一即可，后续将用conda安装.bz2；pip安装.whl)
☞.bz2文件下载网址.

---

☞ .whl文件下载网址.

2 将下载的文件可放于Anaconda本地目录pkg中。

3 安装下载的本地文件
在anaconda prompt中跳转到存放离线文件的目录

如需转盘，则先运行下指令

使用对应指令进行安装torch和torchvision：（conda安装.bz2；pip安装.whl)
——————文件后缀也得加！！

conda install --use-local pytorch-1.9.0-py3.7_cuda11.1_cudnn8_0.tar.bz2
conda install --use-local torchvision离线包文件名

---

pip install torch-1.9.0+cu111-cp37-cp37m-win_amd64.whl
pip install torchvision离线包文件名

4 以上只是部分所需库的安装准备，最后需再次运行官网中Run this Command 中运行指令，完成所有所需库的安装

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

---

pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.8.2+cu111

检验

直接在prompt中，逐条输入以下指令：

python
import torch
torch.__version__
torch.cuda.is_available()

出现下图即为安装成功！

常见问题汇总

1、安装时显示找不到torchaudio
解决：按照上文中离线安装方法，在给出的网站下载相应torchaudio离线安装包，进行离线安装即可。
2、安装时显示找不到cudatoolkit

解决：
方法1：打开 ☞ anaconda网站 ，输入cudatoolkit。

找到下图框选的cudatoolkit类型和相应版本，点击打开。

复制框选中的下载代码，在prompt进行粘贴安装即可。

方法2：
☞☞进入镜像网站

找到对应离线包（.tar.bz2），下载后按照上述方法进行离线安装即可

tips: 安装后记得最后再运行一遍Run this Command 中运行指令哦~~

------tbc-------
有用可以点个大拇指哦 🤭

【作者有话说】
以上内容仅为博主自主整理分享，很多内容也是来源于网络，若有侵权，请私聊告知
大家有任何问题可在评论区讨论交流~