Anaconda环境中PyTorch GPU版本安装（2020.03最新+避坑+全面）_niewzh的博客-程序员秘密

技术标签： Python科学计算

一、前期准备

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快速安装【清华源】

本文链接：https://blog.csdn.net/ScapeD/article/details/104696569

原作者删帖不实内容删帖广告或垃圾文章投诉

智能推荐

Guideline 3.2 - Business_lvmenglong888的博客-程序员秘密

iOS 上架3.2.0 被拒，求大神指点

技术方案解决总结_niu0147的博客-程序员秘密

Android 程序员资源开源项目介绍：一、oschina上的资源汇总http://my.oschina.net/luyao/blog/382330二、其他 http://www.trinea.cn/三、github项目汇总地址：https://github.com/Trinea/android-open-project四、开源项目解析：http://cod

git push时报错：packet_write_wait: Connection to 13.229.188.59 port 22: Broken pipe_ 颜的博客-程序员秘密

git push时出错信息packet_write_wait: Connection to 13.229.188.59 port 22: Broken pipefatal: sha1 file '<stdout>' write error: Broken pipefatal: the remote end hung up unexpectedlyfatal: the remot...

VxWorks开发环境-VxWorks6.8-VxWorks6.9-VxWorks7.0-各个CPU型号的BSP_vxworks支持的cpu_smartvxworks-V2的博客-程序员秘密

vxworks，vxworks下载，VxWorks开发环境，VxWorks6.8，VxWorks6.9，VxWorks7.0，各个CPU型号的，BSP；

接口-AuthenticationEntryPoint_疯狂马铃薯的博客-程序员秘密

AuthenticationEntryPoint简介AuthenticationEntryPoint是Spring Security Web一个概念模型接口，顾名思义，他所建模的概念是:“认证入口点”。它在用户请求处理过程中遇到认证异常时，被ExceptionTranslationFilter用于开启特定认证方案(authentication schema)的认证流程。该接口只定义了一个方法 :void commence(HttpServletRequest request, HttpServlet

PLSQL Developer导入Excel报错，ORA-01036-illegal variable name/number_chenmeng2192089的博客-程序员秘密

excel的表头（第一行）不能是中文，也不能没有表头，否则都会报这个错，解决方法是把表头字段名用英文描述。

随便推点

在iPad利用UTM安装Ubuntu20.04的完整过程+远程操控_ipad安装ubuntu_公子扶苏、的博客-程序员秘密

在ipad上安装Ubuntu

Java维护常量方式的比较——接口、常量类与枚举_java 常量接口_得过且过的勇者y的博客-程序员秘密

而对于枚举，参数接收的是枚举中定义的静态对象（即传入的就是事先存在的、枚举中的对象），即常量值地址唯一（因为其构造函数是私有的，无法通过外部构造出对象），所以只要比较地址即可。而处于不同地址的两个对象是可以相同的，所以对于常量来说，用户传入的参数是自己写的（新创建的）常量，与常量类中定义的常量显然是不同的对象，所以要比较的是内容是否相同而非地址。每个枚举都是通过Class在内部实现的，且所有的枚举值都是publicstaticfinal的。常量能做的，枚举都能做，枚举能做的常量不一定能做。......