Работайте с контейнерами Docker и NGC
Образ готовый для работы с Docker контейнерами c GPU ускорением на базе UEFI загрузчика.
Включает в себя:
Для запуска окружения с Tensorflow-gpu выполните команду: conda activate tf-gpu
Для запуска окружения с Pytorch запустите команду: conda activate pytorch
Для запуска контейнеров с GPU-ускорением используйте флаг --gpus all:
docker run --rm --gpus all ubuntu nvidia-smi