The install command if given here 154 as:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch-nightly
# or
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-nightly -c nvidia

so you would need to change the pytorch and pytorch-nightly channel and package names.