Be sure to add gpu-related commands to the docker run command
, such as:
docker run -dit --gpus 7 nvidia/cuda:11.1.1-cudnn8-runtime-ubuntu20.04 /bin/bash
Be sure to add gpu-related commands to the docker run command
, such as:
docker run -dit --gpus 7 nvidia/cuda:11.1.1-cudnn8-runtime-ubuntu20.04 /bin/bash