Be curious not jugdemental
    Search by
    torch

    pytorch

    Installation on WSL

    # installing an older version of torch

# pip3 install torch==1.4.0 -f https://download.pytorch.org/whl/torch_stable.html


conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch

    For other versions use

    Install CUDA 11.7 on WSL2

    sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove --autoremove nvidia-cuda-toolkit
sudo apt remove --autoremove nvidia-*
sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get autoclean
sudo rm -rf /usr/local/cuda*

echo "deb [signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/ /" | sudo tee /etc/apt/sources.list.d/cuda-wsl-ubuntu-x86_64.list

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600

## latest version

wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-wsl-ubuntu-11-7-local_11.7.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-11-7-local_11.7.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-11-7-local/cuda-B81839D3-keyring.gpg /usr/share/keyrings/

sudo apt-get update
sudo apt-get -y install cuda-11-7
sudo apt install nvidia-cuda-toolkit

sudo apt-get --yes install cuda-toolkit-11-0 cuda-toolkit-10-2


sudo apt install libjpeg-dev zlib1g-dev

pip3 install torch==1.11.0 torchvision==0.12.0 Pillow==9.1.1 --extra-index-url https://download.pytorch.org/whl/cu117

pip3 install --upgrade --force-reinstall --ignore-installed --no-cache-dir  torch==1.11.0 torchvision==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu117

## compatible with pytorch 1.11.0
wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-wsl-ubuntu-11-6-local_11.6.0-1_amd64.deb

sudo dpkg -i cuda-repo-wsl-ubuntu-11-6-local_11.6.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-6-local/7fa2af80.pub

sudo apt-get update
sudo apt-get -y install cuda-11-6
sudo apt install libjpeg-dev zlib1g-dev

sudo apt-get install libfreetype6-dev
sudo apt-get install pkg-config

pip3 install torch==1.13.0.dev20220524+cu116 torchvision==0.13.0.dev20220524+cu116 --extra-index-url https://download.pytorch.org/whl/nightly/cu116

# check installed cuda version
ls -l /usr/local | grep cuda
# or
nvidia-smi

    Install CUDA 11.6 on WSL2

    sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove --autoremove nvidia-cuda-toolkit
sudo apt remove --autoremove nvidia-*
sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get autoclean
sudo rm -rf /usr/local/cuda*

echo "deb [signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/ /" | sudo tee /etc/apt/sources.list.d/cuda-wsl-ubuntu-x86_64.list

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600

## compatible with pytorch 1.11.0
wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-wsl-ubuntu-11-6-local_11.6.0-1_amd64.deb

sudo dpkg -i cuda-repo-wsl-ubuntu-11-6-local_11.6.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-6-local/7fa2af80.pub

sudo apt-get update
sudo apt-get -y install cuda-11-6
sudo apt-get -y install libjpeg-dev zlib1g-dev libfreetype6-dev pkg-config libgdal-dev gdal-bin

pip3 install torch==1.13.0.dev20220524+cu116 torchvision==0.13.0.dev20220524+cu116 --extra-index-url https://download.pytorch.org/whl/nightly/cu116

    nightly builds can be found here

    https://stackoverflow.com/questions/9727688/how-to-get-the-cuda-version

    Install CUDA 10.1 on WSL2 ubuntu 20

    sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove --autoremove nvidia-cuda-toolkit
sudo apt remove --autoremove nvidia-*
sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get autoclean
sudo rm -rf /usr/local/cuda*
sudo apt update
sudo apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC
sudo apt-key adv --keyserver keyserver.ubuntu.com --recv-keys F60F4B3D7FA2AF80
sudo add-apt-repository ppa:graphics-drivers
sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list'
sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda_learn.list'
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt update

sudo apt install cuda-10-2
sudo apt install nvidia-cuda-toolkit
sudo apt install libcudnn7

    based on

    Installation on Windows with conda

    conda update -n base conda

conda uninstall pytorch torchvision torchaudio
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c conda-forge

conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch-nightly -c conda-forge

conda install nomkl

pip install https://download.pytorch.org/whl/nightly/cu116/torch-1.13.0.dev20220612%2Bcu116-cp38-cp38-win_amd64.whl
pip install https://download.pytorch.org/whl/nightly/cu116/torchvision-0.14.0.dev20220612%2Bcu116-cp38-cp38-win_amd64.whl

    Installation on MAC M1

    $ pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu

 pip install https://download.pytorch.org/whl/nightly/cpu/torch-1.13.0.dev20220612-cp38-none-macosx_11_0_arm64.whl
 pip install  https://download.pytorch.org/whl/nightly/cpu/torchvision-0.14.0.dev20220609-cp38-cp38-macosx_11_0_arm64.whl
 pip install https://download.pytorch.org/whl/nightly/cpu/torchvision-0.14.0.dev20220609-cp38-cp38-macosx_11_0_arm64.whl

    CUDA Version

    # check installed cuda version
ls -l /usr/local | grep cuda
# or
nvidia-smi