CUDA的 torch 检测

Question

Closed. This question does not meet Stack Overflow guidelines。它当前不接受答案。












想要改善这个问题吗？更新问题，以便将其作为 on-topic用于堆栈溢出。

7个月前关闭。



Improve this question




哪个 命令可以看到conda env中 pytorch看到的“正确” CUDA版本？ This，是一个类似的问题，但并没有帮助我。

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 ??)(它
好像):

AssertionError: The NVIDIA driver on your system is too old(found version 10010)

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

10020 cuda compiled version

为什么会有这么多不同的版本？我想念什么？
聚苯乙烯
我在带有Geforce 1050的Ubuntu 16.04上安装了Nvidia驱动程序 430。
当我从ubuntu( libcuda1-430)的 additional drivers选项卡安装驱动程序时，使用 Software and Updates。我安装了 pytorch与 conda一起使用 cudatoolkit安装了 conda install -c fastai -c pytorch -c anaconda fastai

最佳答案

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

conda activate myenv

torch.version.cuda

Nvidia-smi仅显示兼容版本。似乎并没有谈论 pytorch自己的cuda所基于的版本。

关于CUDA的 torch 检测，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/64089854/

上一篇：python - Numpy 将数组拆分为具有余数的相同大小的 block

下一篇：pandas - 计算列表中的每个项目在以逗号分隔的值的pandas数据框列中出现的次数

相关文章：

python - 图像聚类 - 在 GPU 上分配内存

python-3.x - torch 错误 "RuntimeError: index out of range: Tried to access index 512 out of table with 511 rows"

deep-learning - PyTorch 等深度学习框架在使用多个 GPU 时如何处理内存？

python - 有没有办法将 torch.nn.DataParallel 与 CPU 一起使用？

python - Numpy/PyTorch - 如何用不同维度的索引分配值？

python-3.x - Wasserstein GAN 批评者训练歧义

pytorch - 名称错误 : name 'nn' is not defined

python - 是否需要在 PyTorch 中清除 GPU 张量？

pytorch - 在 Huggingface Trainer 类中恢复训练时如何避免迭代 Dataloader？

python - PyTorch TensorBoard 写入频率

Answer 1

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

conda activate myenv

torch.version.cuda

Nvidia-smi仅显示兼容版本。似乎并没有谈论 pytorch自己的cuda所基于的版本。