Skip to content

Commit 0c6ac62

Browse files
committed
Testing
1 parent 2c66993 commit 0c6ac62

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

packaging/pkg_helpers.bash

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,10 @@ setup_cuda() {
4444
fi
4545
fi
4646

47+
if [[ "$BUILD_TYPE" == "conda" ]] && [[ "$CU_VERSION" == "cu115"]]; then
48+
export PYTORCH_VERSION_SUFFIX="-py3.7_cuda11.5"
49+
fi
50+
4751
# Now work out the CUDA settings
4852
case "$CU_VERSION" in
4953
cu115)

0 commit comments

Comments
 (0)