如何正确查看pytorch运行时真正调用的cuda版本

这篇具有很好参考价值的文章主要介绍了如何正确查看pytorch运行时真正调用的cuda版本。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

一般情况我们会安装使用多个cuda版本。而且pytorch在安装时也会自动安装一个对应的版本。

正确查看方式: 

想要查看 Pytorch 实际使用的运行时的 cuda 目录,可以直接输出之前介绍的 cpp_extension.py 中的 CUDA_HOME 变量。

import torch
import torch.utils
import torch.utils.cpp_extension

torch.utils.cpp_extension.CUDA_HOME  

上面输出的/usr/local/cuda即为软链接的cuda版本。

如何正确查看pytorch运行时真正调用的cuda版本

 不正确查看方式:

事实上,使用torch,version.cuda命令查看输出的 cuda 的版本并不一定是 Pytorch 在实际系统上运行时使用的 cuda 版本,而是编译该 Pytorch release 版本时使用的 cuda 版本。

如何正确查看pytorch运行时真正调用的cuda版本文章来源地址https://www.toymoban.com/news/detail-508689.html

到了这里,关于如何正确查看pytorch运行时真正调用的cuda版本的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • GPU版本pytorch的安装,配套环境python、Cuda、Anaconda安装和版本选择,及常见问题调用gpu返回false

    前言 :第一次装这个我也很懵,就想记录一下交流经验,这个安装最麻烦的是需要各个 版本 都需要 对应 。我也看了很多教程网上基本上安装都是cpu版本,就官网链接安装下来也是cpu版本,然后就不能调用显卡。 本教程使用python3.9、pytorch1.8.0、Cuda11.2版本、Cudnn8.8.1,这个

    2024年02月03日
    浏览(48)
  • 如何查看自己电脑显卡对应的cuda版本

    第一步: 在电脑桌面点击右键选择NVIDIA控制面板   第二步: 查看显卡驱动版本  比如这里我的驱动版本是472.47,然后去这里查看对应的CUDA版本,如下这个表 从表中可以看出我显卡驱动版本对应最高的cuda版本可去到11.4,同时还可可以在NVIDIA控制面板中点击右下角的 系统信

    2024年02月11日
    浏览(56)
  • 【深度学习环境】如何查看CUDA版本 | 两个CUDA版本各表示什么 | 是否可以在同一设备安装多个CUDA版本

    cuda有两个API:runtime(运行时) API和driver(驱动)API。这两个API都有各自对应的cuda版本。 查看cuda版本的命令如下 nvidia-smi:表示电脑可支持cuda的最高版本 ,(gpu driver installer) nvcc -V:表示当前下载的使用的cuda版本,(CUDA Toolkit installer) CUDA Toolkit(runtime)本质上是一个工具包,所以

    2024年02月15日
    浏览(59)
  • 如何判断自己的电脑里有没有cuda以及查看cuda版本

    CUDA是一种由NVIDIA推出的通用并行计算架构,能够帮助开发人员快捷地创建、测试和部署基于深度学习应用程序。而运行 CUDA 应用程序需要系统至少具有一个支持 CUDA 的显卡和CUDA工具包兼容的驱动程序,这些工具包括 CUDA SDK、 CUDA Studio、 CUDA Testing Kit、 CUDA Software Development Ki

    2024年02月15日
    浏览(54)
  • GPU版本的pytorch安装(显卡为3060ti,如何选择对应的cuda版本)

    显卡为3060ti g6x,操作系统win10 要清楚下面的几个常识 1.GPU和CPU是采用不同架构设计出来的,简单来说,GPU会比CPU多很多计算单元,用于训练网络时,速度比CPU快很多。 2.CUDA是一个计算平台和编程模型,提供了操作GPU的接口。 3.网上很多教程说的安装CUDA其实是指CUDA Toolkit,是

    2024年02月08日
    浏览(48)
  • 详解Windows系统下面如何查看CUDA、cuDNN、Python和各个软件包的版本

    这个题目网络上有很多的讲解,但是 查看CUDA、cuDNN版本 和 查看Python与自身各个软件包 是分开的,且cuDNN版本的查看方式似乎已经过时【截止2023-10-23】。由于自身需要且出于回馈 互联网大学 的目的,将相关内容重新整理在此篇博客中。 方法1:在Windows终端中通过以下3条下命

    2024年02月04日
    浏览(57)
  • 查看电脑cuda版本

    输入NVIDIA搜索即可 出现NVIDIA控制面板 点击系统信息 检查了一下,电脑没用过GPU,连驱动都没有 所以,装驱动…… 选版本,下载 下载后双击打开安装 重新输入 nvidia-smi 显示如下 就这样,成了! 这种感觉很奇妙,奈何,我的论文进展很缓慢,多么希望有一天我也可以这样记

    2024年02月19日
    浏览(42)
  • 【Python】pytorch,CUDA是否可用,查看显卡显存剩余容量

    CUDA可用,共有 1 个GPU设备可用。 当前使用的GPU设备索引:0 当前使用的GPU设备名称:NVIDIA T1000 GPU显存总量:4.00 GB 已使用的GPU显存:0.00 GB 剩余GPU显存:4.00 GB PyTorch版本:1.10.1+cu102 windows先装显卡驱动,再装CUDA10.2,最后装了pytorch。 pip install torch 1.10.1+cu102 torchvision 0.13.1+cu10

    2024年02月10日
    浏览(39)
  • ubuntu下查看cudnn版本+cuda版本

    首先进入usr目录 查找cudnn.h文件 运行如下代码 若无反应,则查找cudnn_version.h文件 (好像因为现在cudnn的版本号已经不在上述文件中了,而是单独有一个文件) 运行如下代码 显示的805则代表下载10.1时候的如下图的8.0.5 查看驱动的driverAPI最高支持版本 查看运行Runtime的API,选择

    2024年04月25日
    浏览(42)
  • 已解决正确查看pip版本、查看pip帮助命令

    已解决(pip查看版本报错)Usage:pip [options] no such option: —verson 粉丝群一个小伙伴想用pip查看版本报错,但是发生了报错(当时他心里瞬间凉了一大截,跑来找我求助,然后顺利帮助他解决了,顺便记录一下希望可以帮助到更多遇到这个bug不会解决的小伙伴),报错代码如下

    2023年04月17日
    浏览(41)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包