Ver código fonte

[SSD/PyT] Update Dockerfile

Update Dockerfile to use PyTorch 21.07 as base
nv-kkudrynski 3 anos atrás
pai
commit
b1fc3c46f5
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      PyTorch/Detection/SSD/Dockerfile

+ 1 - 1
PyTorch/Detection/SSD/Dockerfile

@@ -1,4 +1,4 @@
-ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:21.05-py3
+ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:21.07-py3
 FROM ${FROM_IMAGE_NAME}
 FROM ${FROM_IMAGE_NAME}
 
 
 # Set working directory
 # Set working directory