Closed fire closed 3 years ago
The way to do it to create a new environment and run something similar to:
conda install tensorboard tqdm install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
Mamba solves dependencies faster than conda.
Similar to the current cpu version. Provide an environment.yml for cuda based pytorch.
It is non trivial to switch between gpu and cpu.
https://github.com/pytorch/pytorch/issues/30664
.