A100
裝了 cudatoolkit=11.0 cudnn=8.0 TF=2.4.0
會噴這個
2022-05-02 16:27:27.903055: W tensorflow/stream_executor/gpu/redzone_allocator.cc:314] Internal: ptxas exited with non-zero error code 256, output:
Relying on driver to perform ptx compilation.
Modify $PATH to customize ptxas location.
This message will be only logged once.
可是就開始訓練了
而且跑得還蠻快的 比V100快
這樣到底是有沒有成功啊
查一下stackoverflow有人說這有可能是driver version沒有compat捏
可是我有照這個啊
https://www.tensorflow.org/install/source#gpu
tensorflow-2.4.0 py 3.6-3.8 cudnn 8.0 cuda=11.0
難不成是Build tools的版本問題嗎？
Bazel 3.1.0這東西是要另外裝的嗎？