torch.cuda.is_available () Неверно, даже если установлены правильная версия CUDA и драйвер

Question

Я установил Pytorch 1.5.0, используя conda install pytorch torchvision cudatoolkit=10.2 -c pytorch выходы nvidia-smi: NVIDIA-SMI 440.64.00 Driver Version: 440.64.00 CUDA Version: 10.2

Что касается информации на этом сайте все должно быть в порядке.

Однако CUDA по-прежнему не работает, так как torch.cuda.is_available() возвращает False. Кто-нибудь знает, как я могу это решить?

Нужно ли перезагружать машину?

Answer 1

Похоже, это поврежденный драйвер. Понижение версии и перезагрузка решили проблему.

Answer 2

Обратитесь к этой проблеме

Вы должны добавить дополнительный флаг компоновщика в Linker> Input> Additional Dependencies

-INCLUDE:?warp_size@cuda@at@@YAHXZ

Первоначально Я удалил эту строку, так как она кажется странной, и torch.cuda.is_available() возвращал false, пока я не вставил этот флаг, и теперь он работает нормально.