-
Nicolas Hug authored
* trying stuff * Put flags back? * remove second one just to see * It worked but it's because it's not built by CI. Trying another one * Try using TORCH_CUDA_ARCH_LIST instead * oops * Add new env variable to build/script_env * set TORCH_CUDA_ARCH_LIST for the rest of the CUDA versions * don't pass NVCC_FLAGS venv to conda, let's see if it works Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
226126b8