More than 3 years have passed since last update.

RTX3080/3090系によるTorch/Apexのエラー対応

Posted at 2020-12-23

解決方法

export TORCH_CUDA_ARCH_LIST='8.0+PTX'

Apexのコンパイルを行うと以下のエラーが出る。

nvcc fatal   : Unsupported gpu architecture 'compute_86'

compute_86すなわちsm_86はRTX30x系のことで、Torchはそんなアーキテクチャ知らないよといっている。では、知っている体ですませよう。

export TORCH_CUDA_ARCH_LIST='8.0+PTX'

取り合えずこれで、sm_80系で処理が進む。