nvidia GPU 仅适用于 python2.7

nvidia GPU available only for python2.7

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 已经安装。

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

  1. 寻找您的 docker 图片:
$ docker images
  1. 运行 具有指定图像 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
  1. 要访问 jupyter notebook,复制令牌并将其粘贴到浏览器(在下面的示例中,提供您正确的 ip 地址和令牌。port:8888 可能会有所不同,例如try:8889):
http://YOUR_IP:8888/?token=YOUR_TOKEN_FROM_STEP_2_ABOVE