Код: Выделить всё
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
Pytorch cuda 11.2 сейчас недоступен. (pytorch.org)
Итак, я установил версию 11.1.
(с использованием nvidia-smi)
Версия Cuda
Но она показывает ложь.
Код: Выделить всё
torch.cuda.is_available() >>> False
Насколько мне известно, мне не нужно устанавливать набор инструментов cuda для pytorch
Подробнее здесь: https://stackoverflow.com/questions/675 ... what-to-do
Мобильная версия