win10 GTX1060 安装CUDA+PyTorch GPU

您所在的位置:网站首页 1060用那个驱动 win10 GTX1060 安装CUDA+PyTorch GPU

win10 GTX1060 安装CUDA+PyTorch GPU

2023-08-24 09:07| 来源: 网络整理| 查看: 265

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关掉,不然会一直提示可以操作。



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3