pytorch+vs2015+win10

最新推荐文章于 2023-08-05 22:30:26 发布

小小黑人

最新推荐文章于 2023-08-05 22:30:26 发布

阅读量1.1k

点赞数

分类专栏： pytorch 文章标签： pytorch vs2015 win10 windows cuda

原文链接：https://zhuanlan.zhihu.com/p/54350088

版权

pytorch 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

pytorch+vs2015+win10（一）

安装GPU版本的pytorch

参考：[网页]（ https://zhuanlan.zhihu.com/p/54350088），非常感谢

安装GPU版本的pytorch

首先，查看自己电脑显卡的版本，右击“我的电脑”-“属性”-“设备管理器”-“显示适配器”，就可以看到自己的显卡。一些比较老的显卡，后期是不支持在GPU版本下跑网络的，会出现“no kernel image is available for execution on the device”这样的报错，这种情况最后还得用CPU版本解决问题。

一、前期准备

1.更新nvidia驱动
链接: 官网.，前面检查了自己显卡的版本，这里选择对应的进行更新。（Notebooks表示笔记本）
在这里插入图片描述
2.CUDA9的安装
链接：官网
(1)这里以9.2版本为例，按照深绿色进行相应选择，最后点击“Download”。
(2)下载完成后双击安装包，选择自定义安装选项，只勾选CUDA组件即可。
(3)下一步，安装完成。
这里以9.2版本为例说明
3.cuDNN的下载安装
链接：官网
(这个安装包需要注册后，登录才能下载）
(1)选择“Download duDNN”-“login in ”进行注册，一般邮箱即可，然后登录
(2)“同意条款内容”-选择对应自己电脑的版本，比如windows10。
(3)选择和刚才CUDA相对应的版本（比如，上面第2步下载的CUDA9.2），这里“Download cuDNN v7.6.2 for CuDA 9.2”的即可。
(4)解压缩至一个文件夹，给win10系统环境变量-Path-编辑-新建即可，比如：F:\DL\cudnn\cuda\bin
在这里插入图片描述

二、基于Anaconda安装pytorch

1.下载官网
在这里插入图片描述
(1)选择对应操作系统版本的进行下载（windows)，python3.7的即可。

(2)下载完成后，创建自己的python环境：点击 Anaconda Prompt ，进入到类似windows的cmd命令行解释器中，输入以下命令，创建一个独立的运行环境

conda create -n pytorch_gpu pip python=3.6

(3)pytorch_gpu 为环境名字，可自定义，python3.6为该环境中的版本，与安装Anaconda时默认送的python互不干扰,创建过程中会出现确认信息，输入"y"回车。
(4)进入我们创建好的环境中，并安装相应工具包,接下来输入一下命令

conda install pytorch torchvision cuda90 -c pytorch

（这里的duda90对应于刚才下载的CUDA9.X版本）
(5)整个过程会自动安装，同时，也会出现确认界面，同上输入“y”回车即可。
2. 验证pytorch
点击 “Anaconda Prompt”,进入到命令行
（1）进入自己刚才创建的python环境，输入activate pytorch_gpu
（2）输入“python”
（3）输入：

import torch
x = torch.rand(4,3)
print(x)

出现结果如下，表明安装pytorch成功了！
在这里插入图片描述
3.验证GPU驱动和CUDA可以支持pytorch的加速计算
（1）还是刚才环境，输入如下，返回True,说明一切OK了!

三、使用

1.第一次使用时，点击“开始”-“Anaconda Navigator"，先在左上角这里切换环境，
在这里插入图片描述
2.对应的jupyter，直接点击launch，就可以使用了
3.如果想使用Spyder，在其下面选择“Install”即可，然后点击launch即可。
4.以后再次使用就可以直接在“开始”-“Anaconda”下面直接找到了。