установка CuPy с помощью Pytorch - PullRequest
0 голосов
/ 04 марта 2020

Я хочу передать тензоры из Pytorch в CuPy и обратно, чтобы выполнить некоторые операции, недоступные в Pytorch. Каков рекомендуемый способ их совместной установки?

Я уже установил Pytorch с помощью conda и установил пакет cudatoolkit. Я бы предпочел использовать Conda для установки Cupy. Однако инструкция по установке CuPy рекомендует удалить cudatoolkit (в самом низу).

Итак, как мне убедиться, что Pytorch и CuPy используют одну и ту же версию cuda, а также CuDNN, NV CC, et c? Или я не должен использовать conda для установки Cupy?

1 Ответ

1 голос
/ 04 марта 2020

Просто сделайте conda install -c conda-forge cupy и оно должно работать. Эта часть документации устарела и подлежит обновлению .

...