Hongcheng Wang
Hongcheng Wang
you can use ctrl+c to shutdown it .
export TORCH_CUDA_ARCH_LIST="7.5" before training
if not use, please writer export TORCH_CUDA_ARCH_LIST="7.5" in terminal. The "8.0" usually represents the NVIDIA Ampere architecture, which includes a range of GPU models, such as the NVIDIA A100.You can...
i use rtx3090,i input "export TORCH_CUDA_ARCH_LIST="8.0"can slove this "nvcc fatal : Unsupported gpu architecture 'compute_86'" problem.
哈哈哈祝大佬早日毕业~
终端输一次就行，不用改全局环境，不然你的其他代码可能运行不了，先在终端输入export TORCH_CUDA_ARCH_LIST="8.0"回车后在输入python train.py