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的Flask框架应用程序实现使用QQ账号登录的方法
Jun 07 Python
Python3.6通过自带的urllib通过get或post方法请求url的实例
May 10 Python
详解多线程Django程序耗尽数据库连接的问题
Oct 08 Python
Django中更改默认数据库为mysql的方法示例
Dec 05 Python
解决在pycharm运行代码,调用CMD窗口的命令运行显示乱码问题
Aug 23 Python
Django学习之文件上传与下载
Oct 06 Python
Pytorch模型转onnx模型实例
Jan 15 Python
Win 10下Anaconda虚拟环境的教程
May 18 Python
PyQt5-QDateEdit的简单使用操作
Jul 12 Python
python的setattr函数实例用法
Dec 16 Python
Python爬虫分析微博热搜关键词的实现代码
Feb 22 Python
python基于turtle绘制几何图形
Jun 15 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的session cookie错误
2009/08/09 PHP
php的日期处理函数及uchome的function_coomon中日期处理函数的研究
2011/01/12 PHP
PHP和Mysqlweb应用开发核心技术 第1部分 Php基础-3 代码组织和重用2
2011/07/03 PHP
PHP中集成PayPal标准支付的实现方法分享
2012/02/06 PHP
PHP设计模式之装饰者模式
2012/02/29 PHP
php断点续传之文件分割合并详解
2016/12/13 PHP
JavaScript 三种创建对象的方法
2009/10/16 Javascript
javascript定时变换图片实例代码
2013/03/17 Javascript
jQuery实现简单的日期输入格式化控件
2015/03/12 Javascript
详解AngularJS用Interceptors来统一处理HTTP请求和响应
2017/06/08 Javascript
vue多页面开发和打包正确处理方法
2018/04/20 Javascript
nodejs 简单实现动态html的方法
2018/05/12 NodeJs
js运算符的一些特殊用法
2018/07/29 Javascript
element el-input directive数字进行控制
2018/10/11 Javascript
javascript数组去重方法总结(推荐)
2019/03/20 Javascript
微信小程序的线程架构【推荐】
2019/05/14 Javascript
微信小程序结合Storage实现搜索历史效果
2019/05/18 Javascript
监控微信小程序中的慢HTTP请求过程详解
2019/07/05 Javascript
vue.js中ref及$refs的使用方法解析
2019/10/08 Javascript
python使用PyV8执行javascript代码示例分享
2013/12/04 Python
python查询mysql中文乱码问题
2014/11/09 Python
使用python画社交网络图实例代码
2019/07/10 Python
Python BeautifulSoup [解决方法] TypeError: list indices must be integers or slices, not str
2019/08/07 Python
Python中__repr__和__str__区别详解
2019/11/07 Python
python实现单目标、多目标、多尺度、自定义特征的KCF跟踪算法(实例代码)
2020/01/08 Python
python单元测试框架pytest的使用示例
2020/10/07 Python
Python使用cn2an实现中文数字与阿拉伯数字的相互转换
2021/03/02 Python
HTML5 实现一个访问本地文件的实例
2012/12/13 HTML / CSS
物流经理自我评价
2013/09/23 职场文书
中学生打架检讨书
2014/02/10 职场文书
服装采购员岗位职责
2014/03/15 职场文书
小学一年级数学教学计划
2015/01/20 职场文书
社区禁毒宣传活动总结
2015/05/07 职场文书
2015年政府采购工作总结
2015/05/21 职场文书
学习社交礼仪心得体会
2016/01/22 职场文书
小程序与后端Java接口交互实现HelloWorld入门
2021/07/09 Java/Android