GPUを使用したテンソルフローソリューション

tensorfloe-gpuにはもっと制限があるように感じます

検討された実行可能な方法:

1. minicondaをインストールします(anacondaは機能しません。マシンがインストールされている場合は、アンインストールすることをお勧めします)

2.ベース環境にcondainstall tensorflow = 2.0.0をインストールします    

ここで最新バージョン2.3を試しましたが、機能しませんでした。バージョン2.0.0には、cudatoolkit(10.0.130)とcudnn(7.6.5)が付属します。

より簡潔

次に、vscodeで対応するipythonカーネルを選択します

3.GPUバージョンが利用可能かどうかをテストします

import tensorflow as tf
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")

 

上記の結果が表示された場合、それは利用可能であることが証明されます

おすすめ

転載: blog.csdn.net/weixin_43642491/article/details/108817568