CUDA_HOME=/usr/local/cuda CUTLASS_PATH=/opt/cutlass TORCH_CUDA_ARCH_LIST="8.0 9.0+PTX" PATH=/opt/openmpi/bin:/usr/lib/jvm/default-java/bin:/usr/local/cuda/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin LD_LIBRARY_PATH=/opt/openmpi/lib:/usr/local/cuda/lib64:/usr/local/nvidia/lib:/usr/local/nvidia/lib64 LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64/stubs