Update run_rocm.sh

mila-iqia · Oct 25, 2024 · cd43ccd · cd43ccd
1 parent baf5304
commit cd43ccd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/article/run_rocm.sh b/scripts/article/run_rocm.sh
@@ -75,7 +75,7 @@ install_prepare() {
 
         # https://github.com/ROCm/jax/releases/tag/rocm-jaxlib-v0.4.30
         pip install https://github.com/ROCm/jax/releases/download/rocm-jaxlib-v0.4.30/jaxlib-0.4.30+rocm611-cp310-cp310-manylinux2014_x86_64.whl
-        pip install https://github.com/ROCm/jax/archive/refs/tags/rocm-jaxlib-v0.4.30.tar.g
+        pip install https://github.com/ROCm/jax/archive/refs/tags/rocm-jaxlib-v0.4.30.tar.gz
 
         pip uninstall torch_cluster torch_scatter torch_sparse -y
         FORCE_ONLY_CUDA=1 pip install -U -v --use-pep517 --no-build-isolation git+https://github.com/rusty1s/pytorch_cluster.git