import torch

  import torchvision

  print(torch.cuda.is_available())

上面的命令只是检测CUDA是否安装正确并能被Pytorch检测到，并没有说明是否能正常使用，要想看Pytorch能不能调用cuda加速，还需要简单的测试一下：

a = torch.Tensor(5,3)

a=a.cuda()
print（a）

一般来讲，输出主要是报48号错误，也就是CUDA的问题，出现这个问题在于硬件的支持情况，对于算力3.0的显卡来说，如果安装了9.0的CUDA就会出现这个问题，解决的办法是退回CUDA8.0，或者更换更加高端的显卡，或者直接从源码编译，并在源码中做相应设置（修改setup.py文件里的TORCH_CUDA_ARCH_LIST，将这个值改成你当前使用的GPU对应算力！），对于最后一种方法，目前还没有尝试，不过，Pytorch确实做得不错，考虑的很周到~

最新文章

1. python 装饰器的理解
2. 使用zxing生成二维码
3. windows下指定格式文件转移
4. ASP.NET MVC搭建项目后台UI框架—2、菜单特效
5. iOS开发路线简述
6. tomcat从 http转成https，并且去掉端口号
7. 夺命雷公狗ThinkPHP项目之----企业网站16之文章批量删除的完成
8. POJ2225+BFS
9. Mysql 中和同to_char 一样用法的函数
10. uva 1291(dp)
11. 我的Python成长之路---第二天---Python基础（8）---2016年1月9日（晴）
12. pm2部署多个nodejs项目配置教程
13. Linux命令kill和signal
14. JAVA爬虫实践（实践三：爬虫框架webMagic和csdnBlog爬虫）
15. 【原】git常用命令笔记
16. 巧用&&和|| 让逻辑代码更简洁，逼格看起来更高一点(玩笑脸)
17. JQuery攻略（六）菜单导航
18. Python3基础 json.loads 解析json格式的数据，得到一个字典
19. Ajax学习---Ajax基础学习 180128
20. UVA12585_Poker End Games

热门文章

1. NW.js使用及打包
2. C#-之属性（2）
3. 证明与计算(1): Decision Problem, Formal Language L, P and NP
4. webpack(一) 配置
5. VS2013下载与安装
6. 【HTML】行内-块级-行块级
7. celery概述
8. django.core.exceptions.ImproperlyConfigured: mysqlclient 1.3.13 or newer is required; you have 0.9.3.
9. python学习day15 模块(重点)
10. hadoop生态之mapReduce-Yarn