安装以下依赖并重新启动:

sudo apt install -y nvidia-docker2
sudo systemctl daemon-reload
sudo systemctl restart docker

使用类似如下命令创建容器:

sudo docker run -it --name gpu-env --gpus all ubuntu

注意加上--gpus all这样的参数即可

Logo

分享最新的 NVIDIA AI Software 资源以及活动/会议信息,精选收录AI相关技术内容,欢迎大家加入社区并参与讨论。

更多推荐