pytorch 查看cuda 版本方式
更新时间：2020年06月23日 09:53:07 作者：峰峰晶晶
这篇文章主要介绍了pytorch 查看cuda 版本方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
由于pytorch的whl 安装包名字都一样，所以我们很难区分到底是基于cuda 的哪个版本。
有一条指令可以查看
import torch
print(torch.version.cuda)
补充知识：pytorch：网络定义参数的时候后面不能加".cuda()"
pytorch定义网络__init__()的时候，参数不能加“cuda()", 不然参数不包含在state_dict()中，比如下面这种写法是错误的
self.W1 = nn.Parameter(torch.FloatTensor(3,3), requires_grad=True).cuda()
应该去掉".cuda()"
self.W1 = nn.Parameter(torch.FloatTensor(3,3), requires_grad=True)
以上这篇pytorch 查看cuda 版本方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇浅谈pycharm的xmx和xms设置方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
这篇文章主要介绍了Python中的lambda和apply用法及说明，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-12-12
这篇文章主要介绍了Python线性方程组求解运算,涉及Python使用scipy.linalg.solve进行方程组求解运算的相关操作技巧,需要的朋友可以参考下2018-01-01
这篇文章主要介绍了python可视化数据plt库实例,下面使用pycharm环境给大家详细介绍，文中提到jupyter和pycharm环境的差别，需要的朋友可以参考下2021-06-06
快速解决Django关闭Debug模式无法加载media图片与static静态文件
这篇文章主要介绍了快速解决Django关闭Debug模式无法加载media图片与static静态文件的操作方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-04-04
这篇文章主要介绍了django 按时间范围查询数据库实例代码，分享了相关代码示例，小编觉得还是挺不错的，具有一定借鉴价值，需要的朋友可以参考下2018-02-02
这篇文章主要介绍了在python开发中，Flaks框架之上在URL链接中添加变量的实现方式，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-09-09
在本篇内容里我们给大家整理了关于python创造虚拟环境的详细方法和步骤，需要的朋友们学习下。2019-03-03
这篇文章主要介绍了详细解读Python字符串的使用与f-string,在 Python 中，引号内的任何内容都是字符串,但是字符串也有很多的用法,需要的朋友一起来看看吧2023-04-04
这篇文章主要介绍了用Q-learning算法实现自动走迷宫机器人的方法示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-06-06
最新评论