win10 GTX1060 安装CUDA+PyTorch GPU

2024-05-19 03:15:46

1.说明

其他博客已经讲得很详细了，但是或多或少会有些问题，我自己安装了一遍发下了如下的一些注意事项，想安装的同学们注意了。
假设你已经安装好了anaconda 和py37。

2.CUDA和 cudnn

安装cuda的时候，大的版本号一定要对，安装cuda的时候会自动帮你更新驱动的版本。例如我现在的显卡支持9.2.1，那么安装cuda9.2.148的时候，会自动让我的显卡驱动更新到9.2.148的版本。
安装的时候用自定义安装，不要选vs studio那个选项，也不要安装vs开发环境(如果只是要用torch的话)。
安装结束后，也不需要编译啥的，直接到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\extras\demo_suite这里去运行band和device开头的程序就行。
cudnn的版本随便选，但一定要选支持cuda9.2的。

3 pytorch

最简单的还是照着其他博主，去官网搜对应的安装语句，但是可能会很慢，有些版本的torch在清华镜像上也找不到，所以建议下载whl直接安装 http://download.pytorch.org/whl/torch_stable.html 选择你对应版本的torchvision(需要的话)和torch。
然后anaconda打开或新建一个虚拟环境，在命令行中pip install “xxx” 即可，xxx代表你下载的两个whl的路径，要写全。
最后运行下面代码就可以啦，会输出对应的版本和true的提示。

import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)

4 问题

看到其他博客下面有人提问，2中的device和band会得到35或者 38的错误代码，这是因为驱动版本不对，或者你的显卡出问题了，可以去修一下，或者退回比较老的显卡驱动版本，安装老的cuda。但是老的cuda未见的支持新的torch。
使用的时候把360关掉，不然会一直提示可以操作。

win10 GTX1060 安装CUDA+PyTorch GPU相关推荐

Tensorflow-gpu安装+Keras安装+Pytorch-gpu安装经验分享/Win10 (conda安装Cuda和Cudnn) 含测试代码
Tensorflow-gpu安装+Keras安装+Pytorch-gpu安装经验分享/Win10 (conda安装Cuda和Cudnn) 含测试代码注意: 本文中的Cuda和Cudnn安装均采用c ...
win10下安装cuda和cudnn
本文电脑: win10 64位家庭中文版 1803版本,i5-7300HQ :显卡:GTX1050i; CUDA版本:cuda_9.0.176_win10.exe (链接:CUDA 9.0) cudn ...
记录Win10正确安装CUDA和cuDNN的过程（记录一些坑）
目录前言 CUDA安装直接安装(失败) 根据驱动程序版本对照安装(失败) 重新安装驱动程序(失败) 再次安装CUDA(失败) 安装CUDA 10.2(成功) cuDNN安装环境配置前言之前电 ...
CentOS7安装cuda及GPU驱动--基于runfile文件
首先要说明的是驱动和cuda是两回事,驱动是介于cuda软件库和硬件之间的程序,在通常安装的cuda Toolkit中是存在驱动的,所以我们通常在安装的时候可以不单独下载驱动,但是软件包中的驱动版本较 ...
win10 server 安装CUDA 9.0
下载cuda9.0,不需要提取安装驱动, 直接安装cuda,自定义选项中有安装驱动选项. 查看版本: nvcc -V cuda9.0下载,可以选择系统 https://developer.nvidia ...
怎样查看cudnn版本_ubuntu16.04+Geforce GTX1060安装CUDA和cuDNN
DELL G7系列,ubuntu16.04,本机显卡型号: GeForce GTX 1060. 主要步骤如下: 一,安装NVIDIA驱动. 1,本笔记本是F12进入BIOS,将secure boot设 ...
Win10+GTX1050Ti安装CUDA和cuDNN的记录
最近机器学习的课设需要用到Tensorflow来跑代码. 有CPU和GPU两个版本的Tensorflow(Google怎么这么优秀,气哭.jpg) 电脑是GTX1050Ti的显卡,4G.系统是Win1 ...
win10下安装CUDA和CUDNN的版本对应
版本对应首先查看你的GPU支持的CUDA版本号:https://blog.csdn.net/Candy_GL/article/details/79435151 根据这个版本号,选择与CUDA对应的t ...
Win10+Linux安装CUDA和CUDNN（详版）
一.Windos安装 1.确定本机显卡型号首先我们要确定本机是否有独立显卡.在计算机-管理-设备管理器-显示适配器中,查看是否有独立显卡,如下图所示,可以看到,当前系统拥有NVIDIA显卡,型号 ...

最新文章

热门文章