File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 2727 export CUDA_ROOT=/usr/local/cuda-11.8
2828 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
2929 export CUDACXX=/usr/local/cuda-11.8/bin/nvcc
30- nvcc --version
31- ls -l $CUDACXX
3230 # For Windows, install CUDA
3331 CIBW_BEFORE_ALL_WINDOWS : |
3432 cd src/deepwave
Original file line number Diff line number Diff line change @@ -51,9 +51,20 @@ test-requires = "pytest scipy torch"
5151test-command = " pytest {project}/tests"
5252# Don't bundle CUDA libraries with the wheel
5353repair-wheel-command = " auditwheel repair -w {dest_dir} {wheel} --exclude libcuda.so.1 --exclude 'libcudart.so.*'"
54+ environment = {
55+ PATH =" $PATH:/usr/local/cuda-11.8/bin" ,
56+ CUDA_HOME =" /usr/local/cuda-11.8" ,
57+ CUDA_ROOT =" /usr/local/cuda-11.8" ,
58+ LD_LIBRARY_PATH =" $LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64" ,
59+ CUDACXX =" /usr/local/cuda-11.8/bin/nvcc"
60+ }
5461
5562[tool .cibuildwheel .windows ]
5663test-command = " pytest {project}/tests/test_scalar.py"
64+ environment = {
65+ CUDA_PATH =" C:\\ Program Files\\ NVIDIA GPU Computing Toolkit\\ CUDA\\ v12.4" ,
66+ PATH =" $PATH:C:\\ Program Files\\ NVIDIA GPU Computing Toolkit\\ CUDA\\ v12.4\\ bin"
67+ }
5768
5869[tool .cibuildwheel .macos ]
5970test-command = " pytest {project}/tests/test_scalar.py"
You can’t perform that action at this time.
0 commit comments