Setup - CuPy Updates (#1231)

ROCm · Nov 27, 2023 · c171c03 · c171c03
1 parent 8ad70f7
commit c171c03
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/MIVisionX-setup.py b/MIVisionX-setup.py
@@ -30,7 +30,7 @@
 __author__ = "Kiriti Nagesh Gowda"
 __copyright__ = "Copyright 2018 - 2023, AMD ROCm MIVisionX"
 __license__ = "MIT"
-__version__ = "2.5.6"
+__version__ = "2.5.7"
 __maintainer__ = "Kiriti Nagesh Gowda"
 __email__ = "mivisionx.support@amd.com"
 __status__ = "Shipping"
@@ -452,14 +452,16 @@
         # CuPy Install
         os.system('sudo -v')
         os.system(linuxSystemInstall+' update')
+        os.system('pip install scipy==1.9.3 cython==0.29.*')
+        os.system('pip install git+https://github.com/ROCmSoftwarePlatform/hipify_torch.git')
         if "Ubuntu" in platfromInfo:
             os.system('sudo '+linuxFlag+' '+linuxSystemInstall +
                       ' '+linuxSystemInstall_check+' install -y git g++ hipblas hipsparse rocrand hipfft rocfft rocthrust-dev hipcub-dev python3-dev')
         else:
             os.system('sudo '+linuxFlag+' '+linuxSystemInstall +
                       ' '+linuxSystemInstall_check+' install -y git g++ hipblas hipsparse rocrand hipfft rocfft rocthrust-devel hipcub-devel python3-devel')
         os.system('sudo -v')
-        os.system('(cd '+deps_dir+'; git clone -b v12.2.0 https://github.com/ROCmSoftwarePlatform/cupy.git; export CUPY_INSTALL_USE_HIP=1; export ROCM_HOME=/opt/rocm; cd cupy; git submodule update --init; pip install -e . --no-cache-dir -vvvv)')
+        os.system('(cd '+deps_dir+'; git clone -b rocm6.0_internal_testing https://github.com/ROCmSoftwarePlatform/cupy.git; export CUPY_INSTALL_USE_HIP=1; export ROCM_HOME=/opt/rocm; cd cupy; git submodule update --init; pip install -e . --no-cache-dir -vvvv)')
         os.system('pip install numpy==1.21')
 
     # Install ffmpeg

diff --git a/README.md b/README.md
@@ -444,7 +444,7 @@ Review all notable [changes](CHANGELOG.md#changelog) with the latest release
 * RPP - [1.4.0](https://github.com/GPUOpen-ProfessionalCompute-Libraries/rpp/releases/tag/1.4.0)
 * FFMPEG - [n4.4.2](https://github.com/FFmpeg/FFmpeg/releases/tag/n4.4.2)
 * Dependencies for all the above packages
-* MIVisionX Setup Script - `V2.5.6`
+* MIVisionX Setup Script - `V2.5.7`
 
 ### Known issues