Установите Pytorch GPU с предустановленными CUDA и cudnn

Question

Как видно из названия, я предварительно установил CUDA и cudnn (мой Tensorflow использует их).

Версия CUDA: 10.0 от nvcc --version.

Версия cudnn: 7.4 .

Я пытаюсь установить pytorch в среде conda, используя conda install pytorch torchvision cudatoolkit=10.0 -c pytorch.

Однако установленный pytorch не может успешно определить мой графический процессор.

Кто-нибудь знает, есть ли способ установить Pytorch с GPU-версией с указанными c версиями CUDA и cudnn? Я не хочу менять версию CUDA и cudnn, потому что мой Tensorflow использует их.

Буду признателен за любые идеи!

Answer 1

Так что я наконец решил это сам. Проблема в том, что я не перезагружал свою систему после установки pytorch. После перезагрузки torch.cuda.is_available() возвращает True, как и ожидалось.