nvidia GPU 仅适用于 python2.7

Posted 2023-03-29

【中文标题】nvidia GPU 仅适用于 python2.7

【英文标题】：nvidia GPU available only for python2.7

【发布时间】：2020-09-18 10:02:48

【问题描述】：

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

【参考方案1】：

我能够通过使用 nvidia docker images 解决这个问题

查找您的 docker 映像：

$ docker images

使用指定的 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

要访问 jupyter notebook，复制令牌并将其粘贴到浏览器上（在下面的示例中，提供正确的 ip 地址和令牌。如果端口：8888 已经被占用，则可能会有所不同，例如 try:8889） ：

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