Win10+GPU版Pytorch1.1安装的安装步骤


Posted in Python onSeptember 27, 2019

安装cuda

更新nvidia驱动

打开GeForce Game Ready Driver或在GeForce Experience中下载符合自己gpu的程序。

Win10+GPU版Pytorch1.1安装的安装步骤

选择cuda

打开nvidia控制面板

Win10+GPU版Pytorch1.1安装的安装步骤

点击帮助、点击系统信息、在点击组件在3D设置中可以看到cuda信息

Win10+GPU版Pytorch1.1安装的安装步骤

在我升级过nvidia驱动后,cuda的版本更新到了10.1。接下来下载cuda .

cuda10.1安装完毕。

安装cuDNN

在安装了cuda10.1后选择对应的cuDNN版本v7.6.1

解压文件,然后添加bin目录到环境变量

安装cuDNN

在安装了cuda10.1后选择对应的cuDNN版本v7.6.1

解压文件,然后添加bin目录到环境变量

Win10+GPU版Pytorch1.1安装的安装步骤

使用Anaconda prompt创建环境

conda create -n pytorch_gpu pip python=3.7

其中pytorch_gpu是环境的名字,python版本是3.7

Win10+GPU版Pytorch1.1安装的安装步骤

conda activate pytorch_gpu

这个命令可以进入创建好的环境。

安装Pytorch

建议切换源,我使用的是中国科学技术大学”(USTC Open Source Software Mirror)提供anaconda的镜像源,速度提高不少。

方法一:

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

方法二:

打开.condarc文件,进行编辑

Win10+GPU版Pytorch1.1安装的安装步骤

这个顺序存在一个优先级,会先在中国科学技术大学的镜像源寻找,如果大家感觉速度慢,也可以换别的镜像源。

打开官网选择后复制命令

Win10+GPU版Pytorch1.1安装的安装步骤

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

Win10+GPU版Pytorch1.1安装的安装步骤

整个过程只需要按一下'y'即可。

验证

为了确保PyTorch是否安装成功,我们需要运行简单的样例代码测试,例如打印出随机生成的张量矩阵,以及gpu是否可以使用。

import torch
x = torch.rand(5,5)
print(x)

输出类似下面

tensor([[0.7078, 0.1424, 0.3411, 0.3987, 0.3476],
    [0.7534, 0.7137, 0.3489, 0.4226, 0.3640],
    [0.4104, 0.8411, 0.5112, 0.0629, 0.0664],
    [0.7568, 0.9495, 0.3300, 0.2392, 0.6441],
    [0.7615, 0.1883, 0.6001, 0.9663, 0.3313]])

如果ok,pytorch安装成功。

torch.cuda.is_available()

返回True,GPU版Pytorch安装完毕。

将PyTorch导入Pycharm

点击Pycharm的选项栏中的File选择Settings,然后点击Project Interpreter,接着点击右侧的Add。

Win10+GPU版Pytorch1.1安装的安装步骤

然后在Conda Environment中选择Existing environment

选择Anaconda3下的python.exe,Conda executable选择conda.exe

Win10+GPU版Pytorch1.1安装的安装步骤

如果失败,选择Anaconda3下的env下的新创建的虚拟环境中选择python,试一试。

Pycharm开启科学计算模式(Scientific Mode)

具体步骤:

  1. Settings ?> Tools ?> Python Scientific > Show plots in tool window 勾选
  2. View 勾选 Scientific Mode
  3. Run?>Edit configurations…
  4. 勾选 Run with python console

Win10+GPU版Pytorch1.1安装的安装步骤

开启科学运算模式,舒服,正在习惯这个姿势。

ps:如果想恢复到普通模式,只需要在view下,取消scientific mode勾选。

结束语

​ 环境安装完毕,以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python 网络爬虫--关于简单的模拟登录实例讲解
Jun 01 Python
Python使用字典的嵌套功能详解
Feb 27 Python
python实现微信防撤回神器
Apr 29 Python
Django 拆分model和view的实现方法
Aug 16 Python
wxPython色环电阻计算器
Nov 18 Python
Python @property装饰器原理解析
Jan 22 Python
python 实现图片批量压缩的示例
Dec 18 Python
Python实现区域填充的示例代码
Feb 03 Python
python glom模块的使用简介
Apr 13 Python
Python趣味挑战之给幼儿园弟弟生成1000道算术题
May 28 Python
python获取字符串中的email
Mar 31 Python
Python实战实现爬取天气数据并完成可视化分析详解
Jun 16 Python
opencv调整图像亮度对比度的示例代码
Sep 27 #Python
详解Django将秒转换为xx天xx时xx分
Sep 27 #Python
pytorch多GPU并行运算的实现
Sep 27 #Python
Python使用matplotlib 模块scatter方法画散点图示例
Sep 27 #Python
python利用openpyxl拆分多个工作表的工作簿的方法
Sep 27 #Python
Python绘制热力图示例
Sep 27 #Python
python将print输出的信息保留到日志文件中
Sep 27 #Python
You might like
默默小谈PHP&MYSQL分页原理及实现
2007/01/02 PHP
php静态文件返回304技巧分享
2015/01/06 PHP
推荐JavaScript实现继承的最佳方式
2014/11/11 Javascript
js获取字符串最后一位方法汇总
2014/11/13 Javascript
javascript结合CSS实现苹果开关按钮特效
2015/04/07 Javascript
jquery UI Datepicker时间控件的使用方法(加强版)
2015/11/07 Javascript
js获取及修改网页背景色和字体色的方法
2015/12/29 Javascript
jQuery stop()用法实例详解
2016/07/28 Javascript
js正则表达式注册页面表单验证
2016/10/11 Javascript
微信小程序 自己制作小组件实例详解
2016/12/22 Javascript
BootStrap组件之进度条的基本用法
2017/01/19 Javascript
react-redux中connect()方法详细解析
2017/05/27 Javascript
JavaScript实现图片无缝滚动效果
2017/07/07 Javascript
微信小程序页面滚动到指定位置代码实例
2019/09/07 Javascript
[03:06]V社市场总监Dota2项目负责人Erik专访:希望更多中国玩家加入DOTA2
2014/07/11 DOTA
python的random模块及加权随机算法的python实现方法
2017/01/04 Python
Python之Scrapy爬虫框架安装及简单使用详解
2017/12/22 Python
Python实现获取nginx服务器ip及流量统计信息功能示例
2018/05/18 Python
Django中使用第三方登录的示例代码
2018/08/20 Python
Windows 64位下python3安装nltk模块
2018/09/19 Python
python读取txt文件并取其某一列数据的示例
2019/02/19 Python
python fuzzywuzzy模块模糊字符串匹配详细用法
2019/08/29 Python
python实现从ftp上下载文件的实例方法
2020/07/19 Python
BIBLOO波兰:捷克的一家在线服装店
2018/03/09 全球购物
AVI-8手表美国官方商店:AVI-8 USA
2019/04/10 全球购物
Cult Gaia官网:美国生活方式品牌
2019/08/16 全球购物
10条PHP编程习惯
2014/05/26 面试题
教师师德教育的自我评价
2013/10/31 职场文书
自动一体化专业求职信
2014/03/15 职场文书
员工入职担保书范文
2014/04/01 职场文书
跳槽求职信范文
2014/05/26 职场文书
领导批评与自我批评范文
2014/10/16 职场文书
长江三峡导游词
2015/01/31 职场文书
大足石刻导游词
2015/02/02 职场文书
上课睡觉万能检讨书
2015/02/17 职场文书
详细介绍python操作RabbitMq
2022/04/12 Python