一、問題描述
兩個(gè)問題一并解決:
<1>
Traceback (most recent call last):
? File "run_trainer_ernie_gen.py", line 120, in <module>
? ? paddle.set_device(trainer_params.get("PADDLE_PLACE_TYPE", "cpu"))
? File "/opt/conda/envs/ERNIE-GEN/lib/python3.7/site-packages/paddle/device/__init__.py", line 204, in set_device
? ? place = _convert_to_place(device)
? File "/opt/conda/envs/ERNIE-GEN/lib/python3.7/site-packages/paddle/device/__init__.py", line 127, in _convert_to_place
? ? raise ValueError("The device should not be 'gpu', "
ValueError: The device should not be 'gpu', since PaddlePaddle is not compiled with CUDA
?<2>
RuntimeError: (PreconditionNotMet) Cannot load cudnn shared library. Cannot invoke method cudnnGetVersion.
? [Hint: cudnn_dso_handle should not be null.] (at /paddle/paddle/phi/backends/dynload/cudnn.cc:59)
?問題總結(jié):paddlepaddle調(diào)用不了GPU
二、問題解決
很多博客說安裝cudatoolkit=10.2就可以解決,但是我不行(可能因?yàn)槲沂怯玫膁ocker 容器)
如果在本機(jī)上安裝,可以直接conda安裝:
conda install cudatoolkit=10.2
而docker中我選擇直接去官網(wǎng)下載:
https://developer.nvidia.com/rdp/cudnn-archive
下載內(nèi)容為:?
?切記!一定要選用這個(gè)!
下載完畢,移動(dòng)到/usr/local/目錄下:
tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz
sudo cp cuda/lib64/* /usr/local/cuda-10.2/lib
sudo cp cuda/lib64/* /usr/local/cuda-10.2/lib64/
解壓好CUDA后,下面配置path:
export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64$LD_LIBRARY_PATH
這時(shí)候已經(jīng)大功告成了!
但是我這時(shí)候報(bào)這個(gè)錯(cuò)誤:
檢查一圈發(fā)現(xiàn)需要安裝?paddlepaddle-gpu:
pip install paddlepaddle-gpu
大功告成!
成功的樣子:
文章來源:http://www.zghlxwxcb.cn/news/detail-401782.html
?文章來源地址http://www.zghlxwxcb.cn/news/detail-401782.html
到了這里,關(guān)于ValueError: The device should not be ‘gpu‘, since PaddlePaddle is not compiled with CUDA問題解決(Paddle)的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!