Update pegasus.rst

PegasusResearch · Nov 17, 2024 · deb9922 · deb9922
1 parent 03028fe
commit deb9922
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/docs/source/vehicles/pegasus.rst b/docs/source/vehicles/pegasus.rst
@@ -336,6 +336,13 @@ Installing Pytorch
 
       https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html
 
+      # Install CUSparse for accelerating computations
+      wget https://developer.download.nvidia.com/compute/cusparselt/redist/libcusparse_lt/linux-aarch64/libcusparse_lt-linux-aarch64-0.6.3.2-archive.tar.xz
+      tar xf libcusparse_lt-linux-aarch64-0.6.3.2-archive.tar.xz
+      sudo cp -a libcusparse_lt-linux-aarch64-0.6.3.2-archive/include/* /usr/local/cuda/include/
+      sudo cp -a libcusparse_lt-linux-aarch64-0.6.3.2-archive/lib/* /usr/local/cuda/lib64/
+      sudo ldconfig
+
       # Check the versions + links for pytorch + tensorimage stuff
       https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048
       export JP_VERSION=60
@@ -364,4 +371,4 @@ Installing Tensorflow
       export TENSORFLOW_INSTALL=https://developer.download.nvidia.com/compute/redist/jp/v$JP_VERSION/tensorflow/$TF_VERSION
       sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
       python3 -m pip install -U testresources setuptools numpy future mock keras_preprocessing keras_applications gast protobuf pybind11 cython pkgconfig packaging h5py
-      python3 -m pip install --no-cache $TENSORFLOW_INSTALL
+      python3 -m pip install --no-cache $TENSORFLOW_INSTALL