CUDA 的 Pytorch 检测

Question

CUDA 的 Pytorch 检测

哪个命令可以查看pytorchconda env 中看到的“正确”CUDA 版本？这是一个类似的问题，但并没有让我走得更远。

nvidia-smi 说我有 cuda 版本 10.1
conda list 告诉我 cudatoolkit 版本是 10.2.89
torch.cuda.is_available()显示 FALSE，所以它看到No CUDA?
print(torch.cuda.current_device()), 我得到10.0.10(10010??) （看起来像）：

断言错误：您系统上的 NVIDIA 驱动程序太旧（发现版本 10010）
print(torch._C._cuda_getCompiledVersion(), 'cuda compiled version')告诉我我的版本是10.0.20(10020??)？

10020 cuda编译版

为什么有这么多不同的版本？我错过了什么？

PS 我430在带有 Geforce 1050 的 Ubuntu 16.04 上libcuda1-430安装了Nvidia 驱动程序。当我从additional driversubuntu ( Software and Updates) 中的选项卡安装驱动程序时附带它。我安装pytorch 了conda其中还安装了cudatoolkit使用conda install -c fastai -c pytorch -c anaconda fastai

Answer 1

The*_*ran 14

在安装了 pytorch 的 conda env (myenv) 中，执行以下操作：

conda activate myenv

torch.version.cuda

Run Code Online (Sandbox Code Playgroud)

Nvidia-smi只显示兼容版本。好像没有讲版本pytorch自己的cuda是建立在什么之上的。

我的系统中安装了 cuda-10.2 和 cuda-11.1。我为我的环境导出了 cuda-11.1 的 PATH。```which nvcc``` 返回 ```cuda-11.1``` 但 ```torch.version.cuda``` 显示 ```10.2```。可以做什么？ (4认同)
答案：“哪个命令可以查看 conda env 中的 pytorch 所看到的“正确”CUDA 版本？” 将是：`conda activate my_env`，然后是`conda list | grep cuda`。这应该显示 pytorch 使用的 cuda 和 cudnn 的版本。 (3认同)

归档时间：	5 年，2 月前
查看次数：	11594 次
最近记录：	4 年，3 月前