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

Doctor_Wu_

已于 2022-12-13 14:47:46 修改

阅读量5.3k

点赞数 12

分类专栏：深度学习: 环境配置文章标签： pytorch 深度学习 python

于 2022-01-13 16:49:45 首次发布

本文链接：https://blog.csdn.net/boys_wu/article/details/122477355

版权

17 篇文章 0 订阅

订阅专栏

本文介绍了如何正确地查看Pytorch实际使用的CUDA版本及其目录。通过输出cpp_extension.py中的CUDA_HOME变量来确定Pytorch运行时所依赖的CUDA版本，并指出使用torch.version.cuda命令获取的版本可能与实际运行时版本不同。

摘要由CSDN通过智能技术生成

一般情况我们会安装使用多个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        #输出 Pytorch 运行时使用的 cuda

在这里插入图片描述
上面输出的/usr/local/cuda即为软链接的cuda版本。

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

关注

专栏目录