We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c85da84 commit d1cea8bCopy full SHA for d1cea8b
conda/pytorch-nightly/build.sh
@@ -74,7 +74,7 @@ if [[ -n "$build_with_cuda" ]]; then
74
TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;9.0"
75
# for cuda 12.1 we use cudnn 8.8 and include all dynamic loading libraries
76
export USE_STATIC_CUDNN=0
77
- DEPS_LIST=(/usr/local/cuda/lib64/libcudnn*.so.8 /usr/local/cuda-12.1/extras/CUPTI/lib64/libcupti.so.12
+ DEPS_LIST=(/usr/local/cuda/lib64/libcudnn*.so.8 /usr/local/cuda-12.1/extras/CUPTI/lib64/libcupti.so.12)
78
fi
79
if [[ -n "$OVERRIDE_TORCH_CUDA_ARCH_LIST" ]]; then
80
TORCH_CUDA_ARCH_LIST="$OVERRIDE_TORCH_CUDA_ARCH_LIST"
0 commit comments