Skip to content

Commit 80bbec8

Browse files
authored
Fix cuda 11.6
1 parent 6560b54 commit 80bbec8

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

.github/workflows/building-conda.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ jobs:
9696
- name: Publish Conda package
9797
run: |
9898
conda install anaconda-client --yes
99-
anaconda upload --force -u torch-points3d --label main $HOME/conda-bld/*/*.tar.bz2
99+
anaconda upload --force --label main $HOME/conda-bld/*/*.tar.bz2
100100
env:
101101
ANACONDA_API_TOKEN: ${{ secrets.CONDA_TOKEN }}
102102
shell:

conda/torch-points-kernels/build_conda.sh

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,4 +39,8 @@ echo "PyTorch $TORCH_VERSION+$CUDA_VERSION"
3939
echo "- $CONDA_PYTORCH_CONSTRAINT"
4040
echo "- $CONDA_CUDATOOLKIT_CONSTRAINT"
4141

42-
conda build . -c pytorch -c default -c nvidia --output-folder "$HOME/conda-bld"
42+
if [ "${CUDA_VERSION}" = "cu116" ]; then
43+
conda build . -c pytorch -c default -c nvidia -c conda-forge --output-folder "$HOME/conda-bld"
44+
else
45+
conda build . -c pytorch -c default -c nvidia --output-folder "$HOME/conda-bld"
46+
fi

0 commit comments

Comments
 (0)