我已经从https://tensorflow.org/install/docker设置了tensorflow GPU docker。我正在尝试使用Jupyter笔记本中的import os
库在Ubuntu的tensorflow GPU上加载目录。我可以在终端中使用独立的python访问import os
库并获取目录详细信息,但无法在tensorflow gpu上的Jupyter笔记本中获取目录详细信息。
独立Python:
Python 3.6.7 (default, Oct 22 2018, 11:32:17)
[GCC 8.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import os
>>> base_dir="/home/chief/Desktop/ML/data"
>>> print(len(os.listdir(base_dir)))
3
使用tensorflow gpu的Jupyter Notebook:
我正在使用以下命令运行tensorflow gpu:
sudo docker run --runtime=nvidia -it -p 8888:8888 tensorflow/tensorflow:latest-gpu-py3
我想念什么吗?任何帮助表示赞赏。
答案 0 :(得分:1)
sudo docker run --runtime=nvidia -it -p 8888:8888 tensorflow/tensorflow:latest-gpu-py3 -v /home/chief/Desktop/ML/data:/notebooks/ML_data
希望有帮助。
答案 1 :(得分:0)
您缺少尾部的'/'
base_dir =“ / home / chief / Desktop / ML / data /”
答案 2 :(得分:0)
我可以使用以下命令解决此问题:
docker run --runtime=nvidia -it -p 8888:8888 -v /home/chief/Desktop/ML/:/notebooks tensorflow/tensorflow:latest-gpu-py3