Anaconda环境中PyTorch GPU版本安装（2020.03最新+避坑+全面）

本文主要是介绍Anaconda环境中PyTorch GPU版本安装（2020.03最新+避坑+全面），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

一、前期准备

1. Anaconda 安装

历史版本anaconda下载，推荐下载3-5.2.0版本，该版本对应python3.6。

ps 下载慢的同学：

链接：https://pan.baidu.com/s/1IhKgm5cEPmtj0yos7xpTpw 
提取码：h2ob

检测anaconda是否安装成功：

在开始菜单出，左击 Anaconda Prompt   （如果开始菜单中没有出现此选项，一定安装失败）如果可以成功打开，且左边有 (base)，即安装成功。

2.显卡配置（需要保证有Nvidia 显卡 版本在960M及以上）

首先查询显卡驱动版本：打开命令行输入nvidia-smi

查看版本信息，需要保证大于396.26

若小于396.26，应先更新驱动。推荐使用驱动精灵 更新显卡驱动，更加方便快捷

3.使用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=10.0

大概等待2h左右安装成功，具体视网络情况而定 。

三、检测是否安装成功

接上一步操作，在Anaconda Prompt中

输入 python，打开python环境

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

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

【参考】

[1] PyTorch环境配置及安装

[2] pytorch快速安装【清华源】

这篇关于Anaconda环境中PyTorch GPU版本安装（2020.03最新+避坑+全面）的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---