Every time I have to set up a clean system I manage to mess up cuda somehow, so leaving this as an aide memoir. In general, there is a default nvidia-cuda-toolkit package that ships with Ubuntu-based systems, and you should ignore that. The right options are either:
- conda install cudatoolkit
- This gives you the same thing in a conda env, but is somewhat limited in terms of the range of versions available
- nvidia-container-toolkitt
- Lets you run different cuda versions in docker containers, which can work for having completely isolated environments.
The conda option generally works for me, but do try and install this first as you will end up in a painful package resolution process if you forget and have installed PyTorch first.
Leave a Reply