用的3090跑的,3090算力是8.6,但是目前CUDA为11.0,cuda11.0对应的算力是8.0冲突了
修改环境变量:
sudo vim ~/.bashrc
export TORCH_CUDA_ARCH_LIST=8.0#配置文件中最后一行加这个
source ~/.bashrc
用的3090跑的,3090算力是8.6,但是目前CUDA为11.0,cuda11.0对应的算力是8.0冲突了
修改环境变量:
sudo vim ~/.bashrc
export TORCH_CUDA_ARCH_LIST=8.0#配置文件中最后一行加这个
source ~/.bashrc