# CUDA 11.6
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
# CUDA 11.3
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
# CUDA 10.2
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
# CPU
conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
(pytorch)$ conda install -y pytorch torchvision -c pytorch
(pytorch)$ python -m ipykernel install --user --name pytorch --display-name "PyTorch"