Paddle 我安装了nvidia-docker,为什么训练的时候使用的是CPU,怎么使用GPU资源

apeeds0o  于 2022-04-21  发布在  Java
关注(0)|答案(7)|浏览(180)
  • No description provided.*
wlp8pajw

wlp8pajw1#

请使用nvidia-smi命令确认GPU是否可用,然后确认是通过pip install paddlepaddle-gpu 安装了gpu版本的paddle。

icomxhvb

icomxhvb2#

请使用nvidia-smi命令确认GPU是否可用,然后确认是通过pip install paddlepaddle-gpu 安装了gpu版本的paddle。

都是正常安装的,我宿主机可以使用GPU训练,但是docker不行

czq61nw1

czq61nw13#

请使用nvidia-smi命令确认GPU是否可用,然后确认是通过pip install paddlepaddle-gpu 安装了gpu版本的paddle。

都是正常安装的,我宿主机可以使用GPU训练,但是docker不行

看环境可以使用GPU的,在训练的时候,使用CUDA place就行了

place=paddle.CUDAPlace(0) if core.is_compiled_with_cuda()
else paddle.CPUPlace()

bnl4lu3b

bnl4lu3b4#

请使用nvidia-smi命令确认GPU是否可用,然后确认是通过pip install paddlepaddle-gpu 安装了gpu版本的paddle。

都是正常安装的,我宿主机可以使用GPU训练,但是docker不行

docker里面调用nvidia-smi也正常?

szqfcxe2

szqfcxe25#

请使用nvidia-smi命令确认GPU是否可用,然后确认是通过pip install paddlepaddle-gpu 安装了gpu版本的paddle。

都是正常安装的,我宿主机可以使用GPU训练,但是docker不行

docker里面调用nvidia-smi也正常?

dockers里面不行

vhmi4jdf

vhmi4jdf6#

请使用nvidia-smi命令确认GPU是否可用,然后确认是通过pip install paddlepaddle-gpu 安装了gpu版本的paddle。

都是正常安装的,我宿主机可以使用GPU训练,但是docker不行

docker里面调用nvidia-smi也正常?

dockers里面不行

嗯看样子docker里面没有gpu的环境,你需要通过nvidia-docker run命令启动docker,Paddle已经提供了ubuntu cuda9和cuda10的docker镜像了 https://www.paddlepaddle.org.cn/

5lwkijsr

5lwkijsr7#

请使用nvidia-smi命令确认GPU是否可用,然后确认是通过pip install paddlepaddle-gpu 安装了gpu版本的paddle。

都是正常安装的,我宿主机可以使用GPU训练,但是docker不行

docker里面调用nvidia-smi也正常?

dockers里面不行

嗯看样子docker里面没有gpu的环境,你需要通过nvidia-docker run命令启动docker,Paddle已经提供了ubuntu cuda9和cuda10的docker镜像了 https://www.paddlepaddle.org.cn/

用paddle提供的cuda10的镜像,输nvidia-smi也不行

相关问题