Update Deep-learning.rst

settle · Jun 30, 2020 · 21d6876 · 21d6876
1 parent 29c4b2e
commit 21d6876
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Deep_learning/Deep-learning.rst b/Deep_learning/Deep-learning.rst
@@ -391,8 +391,9 @@ Option 2: Install using Caffe2 ROCm docker image:
 
 ::
 
-  docker pull rocm/pytorch: rocm3.5_ubuntu16.04_py3.6_caffe2
-  docker run -it --network=host --device=/dev/kfd --device=/dev/dri --group-add video -v $PWD:/pytorch rocm/pytorch:rocm3.0_ubuntu16.04_py3.6_caffe2
+  docker pull rocm/pytorch:rocm3.5_ubuntu16.04_py3.6_caffe2
+  docker run -it --network=host --device=/dev/kfd --device=/dev/dri --group-add video -v $PWD:/pytorch rocm/pytorch:rocm3.5_ubuntu16.04_py3.6_caffe2
+
 
 3. Build Caffe2 from source
 ::