jaxlib.xla_extension.XlaRuntimeError: INTERNAL: Failed to execute XLA Runtime executable: run time error: custom call 'xla.gpu.custom_call' failed: ptxas exited with non-zero error code 65280, output: ptxas /tmp/tempfile-taurusi8032-f5aaecd1-56106-601761aa90836, line 5; fatal : Unsupported...
试试这个conda install -c nvidia cuda-nvcc。它将向您的环境添加所需的包。
(0 of 50) | | Elapsed Time: 0:00:00 ETA: --:--:--2020-08-17 18:12:21.594666: W tensorflow/stream_executor/cuda/ptxas_utils.cc:116] *** WARNING *** You are using ptxas 9.1.108, which is older than 9.2.88. ptxas 9.x before 9.2.88 is known to miscompi...