Anaconda 安装pytorch

一、使用conda创建新的Python环境 （可选）

推荐创建新的Python环境将PyTorch安装在其中

左击 Anaconda Prompt 打开，输入以下指令创建环境：

conda create -n torchenv python=3.6

conda 是指调用 conda 包，create 是创建的意思，-n 是指后面的名字是环境的名字， torchenv是环境的名字（可以更改成自己喜欢的），python=3.6 是指创建的屋子，是 python3.6 版本。

之后，弹出提示，输入 y，即可安装。

安装成功后，输入以下指令将当前环境置为新创建的环境：

conda activate torchenv

二、PyTorch安装

推荐使用清华源进行安装，优点安装速度快、成功率高，但版本为旧版本（一样用）

接着上面步骤

在 Anaconda Prompt 中，输入以下命令，修改源

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 --set show_channel_urls yes

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

最后，输入以下命令进行安装：

conda install pytorch torchvision cudatoolkit=9.0

参照
pytorch官网

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

三、检测是否安装成功

接上一步操作，在Anaconda Prompt中

输入 python，打开python环境

输入 import torch ，如果没有报错，意味着 PyTorch 已经顺利安装了

输入 torch.cuda.is_available，如果是 True，意味着你可以使用 GPU，如果是 False，意味着只能使用CPU。

import torch; print(torch.__version__)
import torch; print(torch.version.cuda)

四、为虚拟环境创建spyder

在anaconda下创建多个虚拟环境后，怎样用其中一个虚拟环境下spyder运行程序
激活相应的虚拟环境？

conda activate ‘虚拟环境名’

conda install spyder

五、采用清华源临时安装python包

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple seaborn

六、备注

查看电脑cuda

进入NVIDIA控制面板
在上图的搜索框中，已经显示出NVIDIA控制面板，如果读者有显示，则可以忽略第一步，直接点击进入NVIDIA控制面板。
下面2张图演示如何在控制面板中找出NVIDIA控制面板。
首先选择查看方式为小图标，如下图红色方框标注处所示：