vpuhoff / google_colab_docker_gpu

Configuration of a docker container with a composition as close as possible to the composition of google colab dependencies and support for nvidia cuda gpu
GNU General Public License v3.0
2 stars 0 forks source link

which command to use #1

Open playdasegunda opened 2 years ago

playdasegunda commented 2 years ago

Sorry I'm new to how to use docker, what command should I use to download and install this image?

docker run --gpus=all -it --rm -p 8888:8888 nvcr.io/nvidia/pytorch:21.05-py3

Would it be something like this?

If you can pass the correct command, I would appreciate it, because I just tried to install another local colab in docker, but this problem "NOTE: The SHMEM allocation limit is set to the default of 64MB"...

I would be grateful if you can help,

Yours sincerely, Lucas Rodrigues.

vpuhoff commented 2 years ago

Sorry I'm new to how to use docker, what command should I use to download and install this image?

docker run --gpus=all -it --rm -p 8888:8888 nvcr.io/nvidia/pytorch:21.05-py3

Would it be something like this?

If you can pass the correct command, I would appreciate it, because I just tried to install another local colab in docker, but this problem "NOTE: The SHMEM allocation limit is set to the default of 64MB"...

I would be grateful if you can help,

Yours sincerely, Lucas Rodrigues.

Hi, try to add flag --ipc=host, its should help