pytorch和cuda的版本适配问题

import torch

print(torch.__version__)#注意是双下划线


>>>import torch

>>>torch.version.cuda#输出一个 cuda 版本


torch.cuda.is_available()

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容

  • pytorch笔记
    中文版官方教程：http://pytorch123.com/ 1.一篇总结很好的文章 pytorch学习：http...
    阳光果粒C阅读 1,215评论 1赞 9
  • Pytorch tutorial代码
    Tensors #Tensors#Tensors和numpy中的ndarrays较为相似,因此Tensor也能够使...
    MiracleJQ阅读 1,665评论 0赞 1
  • 帘外飞花掩灯影，落雪只匆匆。
    春雪 夜阑深，人行寂。帘外飞花掩灯影，落雪只匆匆。 车声喧，展印痕。沿踪寻梅难踏雪，阶下水淙淙。 2011年2月2...
    逸川阅读 264评论 0赞 6
  • 遇寺有感
    敬而远，是我唯一能想出来的话。但是敬不代表信仰，信也不能代表信仰。我向来不信仰佛道之说。说白了就是一种心灵寄...
    北温终别南阅读 296评论 0赞 0