python3.6 with pytorch-1.4, tensorflow-1.15, also for cuda 10.0, cudnn7, detectron2, jupyter-lab
147
Run docker with GPU
DEVICES=$(\ls /dev/nvidia* | xargs -I{} echo '--device {}:{}')
docker run -it $DEVICES --shm-size=8096m --net host --env NVIDIA_VISIBLE_DEVICES=0,1 -v /usr/lib64:/usr/lib64 -v /usr/local/cuda-10.0/:/usr/local/cuda/ -p 6600:6600 REPO_NAME:TAG
Content type
Image
Digest
Size
4.3 GB
Last updated
about 6 years ago
docker pull jiajiageng/tensorflow:detectron2