检查conda环境安装了什么cuda
【代码】检查conda环境安装了什么cuda。
·
在虚拟环境中测试 cuda 是否安装成功不能用 nvcc -V。这是因为该命令返回的系统默认的 cuda 版本，而非当前环境下使用的 cuda 版本。我们可以在 python 环境下用 pytorch 命令检查当前环境下执行 python 文件时调用的是哪个 cuda。
import torch
torch.__version__ # 查看 torch 版本
torch.version.cuda # 若 cuda 安装成功则返回版本号，否则失败。
torch.backends.cudnn.version() 
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
1
0- 0
已为社区贡献3条内容
所有评论(0)