Torch.cuda.is_available () продолжает переключаться на False

Question

Я пробовал несколько решений, которые подсказывали, что делать, когда графический процессор CUDA доступен и CUDA установлен, но Torch.cuda.is_available() возвращает False.Они помогли, но только временно, то есть torch.cuda-is_available() сообщили о True, но через некоторое время оно снова переключилось на False.Я использую CUDA 9.0.176 и GTX 1080. Что я должен сделать, чтобы получить постоянный эффект?

Я пробовал следующие методы:

https://forums.fast.ai/t/torch-cuda-is-available-returns-false/16721/5 https://github.com/pytorch/pytorch/issues/15612

Answer 1

1. Установите CUDA 9.1 с помощью apt-get, следуя инструкциям по этой ссылке:

https://cryptoandcoffee.com/mining-gems/cuda-9-0-install-ubuntu-16-04-apt-get/

2. Установлен PyTorch с использованием pip:

   pip install torchvision ( this will install both torch and torchvision )
   

3. Rebooted

Теперь попробуйте:

~$ python -c 'import torch; print torch.cuda.is_available()'