conda 安装 cuda

Install GPU Dependencies

• Nvidia

  conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
  # Alternatively, you can install the nightly version of PyTorch.
  # Install Nightly
  #Install Nightly version (might be more risky)
  conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia