Pytorch检测CUDA

Question

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

1. nvidia-smi说我有cuda版本10.1

2. conda list告诉我 cudatoolkit 版本是10.2.89

3. torch.cuda.is_available()显示 FALSE，所以它看到No CUDA?

4. print(torch.cuda.current_device())，我得到10.0.10（10010??）（看起来像）：

   AssertionError：您系统上的 NVIDIA 驱动程序太旧（发现版本 10010）

5. print(torch._C._cuda_getCompiledVersion(), 'cuda compiled version')告诉我我的版本是10.0.20（10020??）？

   10020 cuda编译版本

为什么有这么多不同的版本？我缺少什么？

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

Answer 1

在安装 pytorch 的 conda env (myenv) 中执行以下操作：conda activate myenv torch.version.cudaNvidia-smi仅显示兼容版本。好像没有讲到pytorch自己的cuda是基于哪个版本构建的。