【问题标题】:nvidia GPU available only for python2.7nvidia GPU 仅适用于 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 已经安装好了。
【问题讨论】:
标签:
python-3.x
tensorflow
gpu
nvidia
torch
【解决方案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