We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 98e4af8 commit c470afeCopy full SHA for c470afe
packaging/pkg_helpers.bash
@@ -53,6 +53,7 @@ setup_cuda() {
53
export CUDA_HOME=/usr/local/cuda-12.1/
54
fi
55
export TORCH_CUDA_ARCH_LIST="5.0+PTX;6.0;7.0;7.5;8.0;8.6;9.0"
56
+ ;;
57
cu118)
58
if [[ "$OSTYPE" == "msys" ]]; then
59
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.8"
0 commit comments