Skip to content

NVIDIA, CUDA, and Pytorch Compatibility

Jump to bottom
Tashrif Billah edited this page Nov 11, 2021 · 2 revisions
  • Pytorch for CUDA 9.1 and NVIDIA 410.xx

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

  • Pytorch for CUDA 9.1 and NVIDIA 397.xx

General guidelines

CUDA 9.2, pytorch 1.7.1 https://pytorch.org/get-started/previous-versions/#linux-and-windows-2 https://download.pytorch.org/whl/cu92/torch_stable.html https://stackoverflow.com/questions/50622525/which-tensorflow-and-cuda-version-combinations-are-compatible CUDA and NVIDIA version compatibilities https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

Clone this wiki locally