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

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/ScapeD/article/details/104696569

智能推荐

使用swift开发OSX应用_703969875的博客-程序员秘密

原文:http://www.raywenderlich.com/87002/getting-started-with-os-x-and-swift-tutorial-part-1翻译原文:http://blog.csdn.net/kmyhy/article/details/45150649打开Xcode,使用 File\NewProject… 菜单,在弹出窗口中选择 “

c++之mutex_c++ mutex_这一切没有想象那么糟的博客-程序员秘密

mutex前言头文件:操作lock, try_lock, unlocklocktry_lockunlocktry_lock_for, try_lock_untillock_guard, unique_lock,std::call_once, std::try_lock, std::lock(批量上锁)lock_guardunique_lockstd::call_oncestd::try_locks...

深度学习:网络优化之梯度下降_影响梯度下降法的因素_DocPark的博客-程序员秘密

应用神经网络模型到机器学习时依然存在一些难点问题. 主要分为两大类:(1)优化问题:神经网络模型是一个非凸函数,很难进行优化。(2)泛化问题:容易在训练集上产生过拟合。因此,目前主要从优化和正则化两个方面来提高学习效率并得到一个好的网络模型。网络优化的问题高维变量的非凸优化低维空间中非凸优化的主要难点是如何选择初始化参数和逃离局部最优点,但在高维而是如何逃离鞍点(Saddle Poin...

小程序 底部按钮兼容 iPhone X(解决底部横杠遮挡问题)_黄河爱浪的博客-程序员秘密

解决方式:底部增加 34px 的高度,使元素不被横杠遮挡。效果图:常规效果 iPhone X 效果 源码:index.wxml<view class="intro"> <view>小程序 iPhone X 底部按钮兼容示例</view> <view>helang.lov...

这些行业月薪过万岗位最多_岗位最多的工作_无忧求职锦囊的博客-程序员秘密

根据前程无忧(www.51job.com)无忧指数的数据库显示,2015年3月全国IT类职能的网上发布职位数逼近60万个,与2014年3月相比增长了29.4%。IT领域尤其以互联网行业为代表一直都是用人大户,即使在2月份受到有农历春节的影响,IT类职能网上发布职位数也仍然保持在55万个左右,进入3月以后,各行业迅速进入招人状态,使得IT类职能网上发布职位数占全国总数的16.3%。  

Effective C++改善程序与设计的55个具体做法笔记_fengbingchun的博客-程序员秘密

Scott Meyers大师Effective三部曲:Effective C++、More Effective C++、Effective STL,这三本书出版已很多年,后来又出版了Effective Modern C++。More Effective C++的笔记见:https://blog.csdn.net/fengbingchun/article/details/102990753E...

随便推点

下拉列表默认选择后台传来的值_下拉选择框直接选定传过来的内容怎么设置_juyuquan1986的博客-程序员秘密

http://www.w3.org/TR/html4/strict.dtd">     Untitled Document     //选中节点的方法,第一个参数为select的id,第二个参数为后台传来的值   function checkOption(id,value)   {    //获得下拉列表的id    var select = document.getElementById(id)

从记事本了解PE结构的导入表_龙图腾的博客-程序员秘密

导入表是杀毒软件厂商非常关注的部位尤其是启发式扫描。所以,了解导入表的结构对免杀技术是必不可少的。 下文我们从大家最常用的notepad.exe的分析来揭开导入表的神秘面纱。 首先用LoadPE工具加载notepad.exe。查看其区段表和目录表信息。  第一步:计算转换因子从目录表的输入表的RVA(相对虚拟地址)查找在区段表中的位置。这里的RVA是0x76

英语不好,不敢学编程,拜托,代码不是死记硬背的_罗夏编程日记的博客-程序员秘密

编程,大家看到源码的一瞬间,这是一堆英文写出来的代码。这里先给大家讲一下,为什么编程用的是英语。编程起源于美国,优先使用了英语。(从机器语言到如今的高级语言);小伙伴可能又有疑问了,计算机发展了这么久,为什么还没出现中文编程;1.输入输出方面,汉字已经完成了搜集(utf-8),基本人机交互可以满足2.想用汉字在原有基础上编程,将汉字转成英文字符再转成机器语言。增加了过程,消耗性能。3.想提高性能,需要重新定义硬件,从机器最基本的与或非开始封装汉字偏旁/注音。下面来点实际的,看看编程需要用到哪些

团队作业3-第二部分-系统设计-2.1系统架构设计_alimoqa1122的博客-程序员秘密

01:开发级需求分析  我们的团队在开始阶段分析了整个系统需要的复杂度等方面,最后确定了开发级需求分析:在开发过程中,我们要考虑到的不仅是这个系统需要满足用户的什么需求,更要根据自身团队的能力和团队的特点,对开发系统的过程做一些开发及的约束。我们作为一个学生团队,时间和精力也比较有限,因此我们对系统开发的效率要求、系统开发的复杂度就必须要有一个标准——在满足用户需求的前提下,...

今天来讲一下Object.prototype.toString.call()、instanceof以及Array.isArray的区别_object.prototype方法放在contructor里和放在外面的区别___潇湘夜雨__的博客-程序员秘密

一、Object.prototype.toString.call()方法Object.prototype.toString.call(undefined);"[object Undefined]"Object.prototype.toString.call(null);"[object Null]"Object.prototype.toString.call('');"[objec...

LDR和HDR_hdr和ldr_Kaitiren的博客-程序员秘密

一、LDR和HDR的基本概念1.HDR 、LDR、动态范围、ToneMapping的基本概念Dynamic Range(动态范围)=最高亮度/最低亮度 HDR= High Dynamic Range LDR= Low Dynamic Range ToneMapping:将超高的动态范围转换到我们日常显示的屏幕上的低动态范围的过程 一些小芝士: 因为不同的厂家生产的屏幕亮度(物理)实际上是不统一的,那么我们在说LDR时,它是一个0到1范围的值,对应到不同的屏幕上就是匹配当前屏幕的最低亮度(0).

推荐文章

热门文章

相关标签