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这东西是要另外装的吗？