torch.cuda.is_available () возвращает false с Ubuntu 16.04 - PullRequest
Новая
       121

torch.cuda.is_available () возвращает false с Ubuntu 16.04

0 голосов
/ 09 мая 2020

У меня следующие конфигурации

  • nv cc -V 
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Tue_Aug_11_14:27:32_CDT_2015
Cuda compilation tools, release 7.5, V7.5.17
  • nvidia-smi

enter image description here

  • Набор инструментов CUDA (от Anaconda)

cudatoolkit 10.2.89 hfd86e86_1 anaconda

  • PyTorch (от Anaconda)

pytorch 1.5.0 py3.7_cuda10.2.89_cudnn7.6.5_0 pytorch

Но я получаю torch.cuda.is_available() -> False

Кто-нибудь, пожалуйста, скажите, какой компонент мне следует обновить или понизить, чтобы использовать CUDA?

1 Ответ

1 голос
/ 11 мая 2020

У меня была такая же проблема на Ubuntu 18.04 с установкой anaconda, рекомендованной с официального сайта PyTorch. После установки cudatoolkit 10.1 вместо рекомендованного 10.2 я получил True. Некоторые люди исправляли это с помощью понижения / обновления версии драйвера, поскольку это, скорее всего, проблема совместимости.

...