Win10+GPU版Pytorch1.1安装

引言

本篇为Pytorch安装。

安装cuda

更新nvidia驱动

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

选择cuda

  1. 打开nvidia控制面板

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

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

    cuda10.1安装完毕。

安装cuDNN

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

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

使用Anaconda prompt创建环境

1
conda create -n pytorch_gpu pip python=3.7

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

1
conda activate pytorch_gpu

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

安装Pytorch

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

方法一:

12
conda config conda config --set show_channel_urls yes

方法二:

打开.condarc文件,进行编辑

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

打开官网选择后复制命令

1
conda install pytorch torchvision cudatoolkit 大专栏  Win10+GPU版Pytorch1.1安装=10.0 -c pytorch

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

验证

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

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

输出类似下面

12345
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安装成功。

1
torch.cuda.is_available()

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

将PyTorch导入Pycharm

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

然后在Conda Environment中选择Existing environment

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

如果失败,选择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

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

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

结束语

? 环境安装完毕,接下来开始我的探索之路,共勉!



Win10+GPU版Pytorch1.1安装

原文地址:https://www.cnblogs.com/lijianming180/p/12032126.html

时间: 2024-10-14 05:03:55

Win10+GPU版Pytorch1.1安装的相关文章

windows 下 TensorFlow(GPU 版)的安装

windows 10 64bit下安装Tensorflow+Keras+VS2015+CUDA8.0 GPU加速 0. 环境 OS:Windows 10,64 bit: 显卡:NVIDIA GeForce GTX 1050Ti,显卡查看方法:计算机[设备管理器]? [显示适配器] Python 的版本,注意只能为 3.5,并非越高越好: CUDA,8.0: keras 可以以 TensorFlow 为后端(backend)(也可以 theano 为后端),安装完成 TensorFlow 的基础上

Win10 + Python + GPU版MXNet + VS2015 + RTools + R配置

最近入手一台GTX 1070的笔记本,手痒想在win10上试下GPU跑模型,所以就有了接下来的安装GPU版mxnet的坎坷历程,经过多重试验终于搞定了python和R安装mxnet,现将主要点记录如下: 本人主要参考这2篇博文: https://my.oschina.net/qinhui99/blog/845249 http://blog.csdn.net/u010414386/article/details/53304177 1.安装vs2013或vs2015 如果安装的是vs2013,则需要

Windows64位安装GPU版TensorFlow 0.12,Power Shell下输入:安装Tensorflow的全教程

推荐使用powershell,只需要在cmd指令窗口输入powershell即可 下载64位Python3.5(一定要3.5!!)可以通过Python 3.5 from python.org 或 Python 3.5 from Anaconda 下载并安装Python3.5.2(注意选择正确的操作系统). 具体教程可以查看Python3.5.2百度经验安装,里面有个细节,自动设置环境变量,不能忘 安装VS2015,如果是2013的后面要加个插件,这是后话 为了使用国内镜像加速pip安装,需要如下

在windows上极简安装GPU版AI框架(Tensorflow、Pytorch)

在windows上极简安装GPU版AI框架 如果我们想在windows系统上安装GPU版本的AI框架,比如GPU版本的tesnorflow,通常我们会看到类似下面的安装教程 官方版本 安装CUDA 安装cuDNN 配置环境变量 安装python环境 安装gpu版的tensorflow开发包 咋看上去好像不是很复杂,但是其中坑多到你怀疑人生. 下载的cuDNN时候需要注册,而且因为cuDNN文件在外网,下载速度很慢. 比如不同版本的tensorflow和CUDA(cuDNN)的版本是有匹配关系的,

在windows上极简安装GPU版AI框架

在windows上极简安装GPU版AI框架 如果我们想在windows系统上安装GPU版本的AI框架,比如GPU版本的tesnorflow,通常我们会看到类似下面的安装教程 官方版本 安装CUDA 安装cuDNN 配置环境变量 安装python环境 安装gpu版的tensorflow开发包 咋看上去好像不是很复杂,但是其中坑多到你怀疑人生. 下载的cuDNN时候需要注册,而且因为cuDNN文件在外网,下载速度很慢. 比如不同版本的tensorflow和CUDA(cuDNN)的版本是有匹配关系的,

收不到Win10正式版预订通知?一个批处理搞定

目前,已经有不少Win7.Win8.1用户在系统右下角收到Win10正式版的预订提示窗口.点击接受预订后,系统会将Win10正式版所需的安装文件提前下载好,7月29日正式发布的时候,就可以第一时间升级了. 不过,也有不少用户反映自己尚未收到提示,外媒ghacks提供了一系列解决办法,大家不妨试试,据说这些方法还是挺有效的. 这里将可能遇到的问题做成了一个批处理文件(点此下载),解压后即可得到win10fix_full.bat文件,以管理员身份运行后,会出现如下提示: 按任意键继续,出现版本列表,

Win10正式版开机慢怎么办 开机黑屏时间长怎么办

升级Win10正式版后开机速度慢.黑屏时间长怎么解决呢?其实我重要是由Win10正式版所提供的“快速启动”功能与电脑显卡驱动.电源管理驱动不兼容所造成的.下面就与大家分享一下针对Win10正式版开机速度慢.黑屏等待时间长的具体解决办法. 方法/步骤 1 首先需要说明一点,对于Win10正式版开机等待时间长的问题,通常只发生在通过“升级”方式更新的Win10正式版系统中.而对于采用“全新”方式安装 的Win10正式版系统中则不会出现此类故障.   接下来就与大家分享一下Win10正式版系统开机速度

Win10正式版系统在哪下载呢

Windows 10 Build 14393.351主要集中在修复系统BUG和改善后台稳定性方面,与往常一样,Win10系统之家利用微软官方映像文件转换成GHO格式镜像,方便使用U盘.硬盘等方法安装,需要的用户可下载使用. 大家会在百度搜索框输入"win10正版","win10官方原版系统"等等,出现的链接也有各种各样的,不排除下载一些盗版系统,所以:请到微软内部网站下载. 1,打开浏览器:搜索msdn, 我告诉你:会出现结果页: 2,选则官网:打开:记住官网的右上

Win10正式版激活方法有哪些?如何激活Win10?

很多用户都想将系统升级到Win10,但是却不知道怎么激活Win10正式版的方法,其实不同版本激活正式版Win10的方法秘钥不同,下面99安卓网小编就分享一些激活Win10正式版的方法和秘钥,供大家参考. 目前win10正式版已经发布,相信已经有非常多的Win7/Win8.1用户升级安装了Win10正式版,对一款系统来说激活相信是大家特别关心的,而微软对win10系统执行了一年免费升级的策略,目前Win7.Win8.1系统只要是在升级Win10系统之前已经激活的,那么你升级安装好Win10系统之后