虽然一直用tensorflow的cpu版,但是每次看到报错信息说发现GPU但没法用很膈应,今天终于找到解决方案:
python3 -m pip install --extra-index-url https://pypi.nvidia.com tensorrt-bindings8.6.1 tensorrt-libs8.6.1
python3 -m pip install -U tensorflow[and-cuda]
我的显卡:Geforce 1050Ti
虽然一直用tensorflow的cpu版,但是每次看到报错信息说发现GPU但没法用很膈应,今天终于找到解决方案:
python3 -m pip install --extra-index-url https://pypi.nvidia.com tensorrt-bindings8.6.1 tensorrt-libs8.6.1
python3 -m pip install -U tensorflow[and-cuda]
我的显卡:Geforce 1050Ti