1
|
export TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6" |
1
|
export MAX_JOBS=32 |
此时需要退出当前的源码目录,可以切换到一个无关的目录,例如:/tmp,然后在新的目录下再次运行python3和import torch指令。