我按照Chainer doc上的說明操作,這導致我在運行代碼時出錯：
RuntimeErrorTraceback (most recent call last)
<ipython-input-9-ffb21f9880f0> in <module>()
...
6 model = Classifier(CompetitionNetwork(n_units = 64))
----> 7 model.to_gpu()
...
RuntimeError: CUDA environment is not correctly set up
(see https://github.com/chainer/chainer#installation).No module named cupy
然后我嘗試用許多不同的方式安裝cupy,其中一個是
!apt -y install libcusparse8.0 libnvrtc8.0 libnvtoolsext1
!ln -snf /usr/lib/x86_64-linux-gnu/libnvrtc-builtins.so.8.0 /usr/lib/x86_64-linux-gnu/libnvrtc-builtins.so
!pip install cupy-cuda80 chainer
匯入cupy然后運行我的代碼后繼續給我同樣的錯誤：
RuntimeError: CUDA environment is not correctly set up (see
https://github.com/chainer/chainer#installation).No module named cupy
接下來我嘗試使用以下方法安裝cuda：
!wget https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64 -O cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb
!dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb
!apt-key add /var/cuda-repo-<version>/7fa2af80.pub
!apt-get update
!apt-get install cuda
這花了很長時間,似乎作業但最終仍然給了我同樣的錯誤.
似乎在Google Colab的GPU上使用Chainer非常困難,除非我做錯了.使用Tensorflow,它更容易.有沒有人有在谷歌GPU上使用Chainer的經驗？
uj5u.com熱心網友回復：
您可能想看看這個Chainer示例.https://colab.research.google.com/drive/1SsxHvQdSz23kaVov8yKizVD3_2tkXdZM
uj5u.com熱心網友回復：
cuda版本為10.0，安裝陳述句：pip install cupy-cuda100 已解決
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/230793.html
標籤：其他技術討論專區
上一篇：pyinstaller 打包 pytorch 出現問題
下一篇：Axure的基礎知識點