電腦：小新pro13，顯卡：MX150，python：3.6.4
安裝CUDA10.2、cuDNN7.6.5
參考：https://blog.csdn.net/m0_37867091/article/details/105788637
安裝pytorch
使用pip安裝 pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html
安裝tensorflow-gpu 1.15.0
1. pip istall tensorflow-gpu==1.15.0
2. 遇到wrapt安裝的錯誤
- 輸入pip install -U --ignore-installed wrapt enum34 simplejson netaddr
![]()
- 再重新安裝pip install tensorflow-gpu==1.15.0
![]()
3. 導入tensorflow時遇到錯誤：找不到dll，Could not load dynamic library 'cudart64_100.dll'
- 把找不到的dll複製一份，重新命名
- 如找不到cudart64_100.dll，把cudart64_102.dll複製一份，重命名爲cudart64_100.dll
結果：可成功使用GPU
