Skip to content

NVIDIA, CUDA, and Pytorch Compatibility

Jump to bottom
Tashrif Billah edited this page Nov 11, 2021 · 2 revisions

Recipes

  • Pytorch for CUDA 9.1 and NVIDIA >= 410.48

https://github.com/tashrifbillah/HD-BET/blob/f7f784bfad7ccc6733703907f1c6bc1179524ebe/environment_gpu.yml

  • Pytorch for CUDA 9.1 and NVIDIA >= 390.46

https://github.com/tashrifbillah/HD-BET/blob/44b3bcb8e4eabb6801ea05b591b82383e3d430bc/environment_gpu.yml

Compatibilities

  • CUDA vs NVIDIA

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

  • Tensorflow vs CUDA

https://stackoverflow.com/a/50622526/11932012

Clone this wiki locally