win10+GPU+pytorch安装教程

Windows10下GPU版本的Pytorch安装教程

1.查看自己的显卡版本型号:
打开显卡控制面板–左下角的系统信息–组件–查看自己的显卡支持的cuda版本型号
2.下载cuda
9.2版本的cuda官网下载:https://developer.nvidia.com/cuda-92-download-archive
选择对应的版本型号如下,并点击download开始下载
在这里插入图片描述
3.下载cudnn:
1.首先需要注册一个账号
在这里插入图片描述
2.官网下载:https://developer.nvidia.com/rdp/cudnn-download
3.选择对应的版本及系统:
在这里插入图片描述
4.安装cuda
按照提示进行即可可以选择默认路径或者自定义
5.测试
测试cuda是否安装成功,找到cuda安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin,cmd打开prompt,cd到指定路径下,执行nvcc -V指令,出现如下结果证明安装成功
在这里插入图片描述
6.cudnn处理
将文件夹解压,并将里面的三个文件夹复制黏贴到cuda安装目录下
在这里插入图片描述
7测试
进入demo文件中进行两个程序的测试C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\extras\demo_suite,执行bandwidthTest.exe及deviceQuery.exe,显示pass则安装成功
在这里插入图片描述
8.pytorch安装
采用conda安装,但是比较慢,预计半个多小时
在这里插入图片描述
将conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev复制 直接conda安装
9.pytorch检测
打开pycharm 输入import Torch 无报错则安装成功。

发布了6 篇原创文章 · 获赞 2 · 访问量 218
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 数字20 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览