Update dependency_check.py

brainpy · Mar 15, 2024 · baec6d7 · baec6d7
1 parent c2b412c
commit baec6d7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/brainpy/_src/dependency_check.py b/brainpy/_src/dependency_check.py
@@ -29,7 +29,8 @@
 numba_install_info = ('We need numba. Please install numba by pip . \n'
                       '> pip install numba')
 cupy_install_info = ('We need cupy. Please install cupy by pip . \n'
-                      '> pip install cupy')
+                      'For CUDA v11.2 ~ 11.8 > pip install cupy-cuda11x\n'
+                      'For CUDA v12.x        > pip install cupy-cuda12x\n')
 os.environ["TI_LOG_LEVEL"] = "error"