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访问sqlserver示例
Feb 10 Python
Python实例之wxpython中Frame使用方法
Jun 09 Python
深入解析Python中的list列表及其切片和迭代操作
Mar 13 Python
Python中的Descriptor描述符学习教程
Jun 02 Python
使用Python读写及压缩和解压缩文件的示例
Jul 08 Python
python制作简单五子棋游戏
Jun 18 Python
Python使用pyserial进行串口通信的实例
Jul 02 Python
python 用所有标点符号分隔句子的示例
Jul 15 Python
python匿名函数用法实例分析
Aug 03 Python
10分钟教你用python动画演示深度优先算法搜寻逃出迷宫的路径
Aug 12 Python
python 调用pyautogui 实时获取鼠标的位置、移动鼠标的方法
Aug 27 Python
QML实现钟表效果
Jun 02 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
磨咖啡豆的密诀
2021/03/03 冲泡冲煮
谨慎使用PHP的引用原因分析
2012/09/06 PHP
浅谈php命令行用法
2015/02/04 PHP
PHP保存带BOM文件的方法
2015/02/12 PHP
判断页面是关闭还是刷新的js代码
2007/01/28 Javascript
学习YUI.Ext 第四天--对话框Dialog的使用
2007/03/10 Javascript
Javascript延迟执行实现方法(setTimeout)
2010/12/30 Javascript
浅析XMLHttpRequest的缓存问题
2013/12/13 Javascript
使用JavaScript获取地址栏参数的方法
2014/12/19 Javascript
js实现透明度渐变效果的方法
2015/04/10 Javascript
Bootstrap布局之栅格系统详解
2016/06/13 Javascript
jQuery+CSS3实现四种应用广泛的导航条制作实例详解
2016/09/17 Javascript
JavaScript中Math对象的方法介绍
2017/01/05 Javascript
JS+html5 canvas实现的简单绘制折线图效果示例
2017/03/13 Javascript
详解Vue.js搭建路由报错 router.map is not a function
2017/06/27 Javascript
JS库之ParticlesJS使用简介
2017/09/12 Javascript
微信小程序promsie.all和promise顺序执行
2017/10/27 Javascript
vue组件文档(.md)中如何自动导入示例(.vue)详解
2019/01/25 Javascript
Vue如何获取数据列表展示
2019/12/11 Javascript
vue-cli3单页构建大型项目方案
2020/04/07 Javascript
浅析PHP与Python进行数据交互
2018/05/15 Python
Python实现的网页截图功能【PyQt4与selenium组件】
2018/07/12 Python
对python numpy.array插入一行或一列的方法详解
2019/01/29 Python
python 实现识别图片上的数字
2019/07/30 Python
使用python实现对元素的长截图功能
2019/11/14 Python
python 发送邮件的四种方法汇总
2020/12/02 Python
联想智利官方网站:Lenovo Chile
2020/06/03 全球购物
党支部书记岗位责任制
2014/02/11 职场文书
法制宣传标语集锦
2014/06/25 职场文书
人大调研汇报材料
2014/08/14 职场文书
企业挂职心得体会
2014/09/10 职场文书
Nginx+SpringBoot实现负载均衡的示例
2021/03/31 Servers
详解python中[-1]、[:-1]、[::-1]、[n::-1]使用方法
2021/04/25 Python
pytorch model.cuda()花费时间很长的解决
2021/06/01 Python
教你怎么用Python实现GIF动图的提取及合成
2021/06/15 Python
mysql分组后合并显示一个字段的多条数据方式
2022/01/22 MySQL