To enable CUDA within docker container

subhash · Mar 1, 2018 · c73509a · c73509a
1 parent 276ce67
commit c73509a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Dockerfile-gpu b/Dockerfile-gpu
@@ -97,8 +97,9 @@ RUN cd /opt/tensorflow/models/research \
 ENV PYTHONPATH=$PYTHONPATH:/opt/tensorflow/models/research:/opt/tensorflow/models/research/slim
 
 # udacity capstone
-RUN mkdir /udacity
-VOLUME ["/udacity"]
+RUN mkdir /capstone
+VOLUME ["/capstone"]
 VOLUME ["/root/.ros/log/"]
 WORKDIR /udacity/
 
+RUN pip install jupyter