Xin Dong
commited on
Commit
·
a98fb88
1
Parent(s):
3d41055
update setup
Browse files
setup.sh
CHANGED
|
@@ -9,6 +9,8 @@ if [[ "$cuda_version" != "12.1" && "$cuda_version" != "12.4" ]]; then
|
|
| 9 |
exit 1
|
| 10 |
fi
|
| 11 |
|
|
|
|
|
|
|
| 12 |
# Install PyTorch with the specified CUDA version
|
| 13 |
conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=$cuda_version -c pytorch -c nvidia
|
| 14 |
|
|
@@ -28,7 +30,7 @@ cd ..
|
|
| 28 |
# Clone and install causal-conv1d with specified CUDA version
|
| 29 |
git clone https://github.com/Dao-AILab/causal-conv1d.git
|
| 30 |
cd causal-conv1d
|
| 31 |
-
|
| 32 |
TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6;8.9;9.0" python setup.py install
|
| 33 |
cd ..
|
| 34 |
|
|
|
|
| 9 |
exit 1
|
| 10 |
fi
|
| 11 |
|
| 12 |
+
export CUDA_HOME=/usr/local/cuda-$cuda_version
|
| 13 |
+
|
| 14 |
# Install PyTorch with the specified CUDA version
|
| 15 |
conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=$cuda_version -c pytorch -c nvidia
|
| 16 |
|
|
|
|
| 30 |
# Clone and install causal-conv1d with specified CUDA version
|
| 31 |
git clone https://github.com/Dao-AILab/causal-conv1d.git
|
| 32 |
cd causal-conv1d
|
| 33 |
+
|
| 34 |
TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6;8.9;9.0" python setup.py install
|
| 35 |
cd ..
|
| 36 |
|