nvidia GPU仅可用于python2.7

Question

enter image description here

nvidia gpu不适用于python3.6，仅适用于python2.7版本。我正在使用ubuntu，cuda 10.1和所有驱动程序正常工作。我也尝试通过创建新的python环境来尝试，但gpu仍可用于python，而不是python3。我应该怎么做才能使用gpu运行python3脚本？

请注意，tensorflow-gpu，Torch已经安装并且可以在python2上运行，唯一的问题是python3不使用gpu，包括jupyter notebook（使用python3）。我使用ssh连接到服务器，已经安装了nvidia docker。

Answer 1

我能够使用nvidia泊坞图像解决问题

1. 寻找您的docker映像：

$ docker images

2. 使用指定的IMAGE ID运行docker映像：

$ docker run --rm -it --runtime=nvidia --net=host -v /<local dir>/:/<destination dir> <IMAGE ID>

$ docker run --rm -it --runtime=nvidia --net=host -v /my_sever_dir/:/notebook 8d78dd1e1q

3. 要访问jupyter笔记本，请复制令牌并将其粘贴到浏览器中（在下面的示例中，提供正确的IP地址和令牌。如果已使用，则端口：8888可能会有所不同，例如try：8889）：

http://YOUR_IP:8888/?token=YOUR_TOKEN_FROM_STEP_2_ABOVE