search

Azure / azhpc-images

Azure HPC/AI VM Images
MIT License
95 stars 77 forks source link

Adding ability to choose ROCm or CUDA SKUs #358

Closed moes1 closed 1 month ago

moes1 commented 2 months ago

A couple changes:

  1. Added "SKU" variable that should be ROCm or CUDA, so we can switch between installing ROCm or CUDA libraries. It is always CUDA except for Ubuntu 22.04 where it must be chosen.
  2. Added tools needed to install ROCm libraries to create a ROCm VM.