Kaynağa Gözat

[SSD/PyT] Update Dockerfile

Update Dockerfile to use PyTorch 21.07 as base
nv-kkudrynski 3 yıl önce
ebeveyn
işleme
b1fc3c46f5
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  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}
 
 # Set working directory