diff --git a/README.md b/README.md index dc4a0093e3d..53038179b75 100644 --- a/README.md +++ b/README.md @@ -124,19 +124,10 @@ pip install --upgrade pip Install PyTorch with CUDA (this repo has been tested with CUDA 11.7 and CUDA 11.8) and [tiny-cuda-nn](https://github.com/NVlabs/tiny-cuda-nn). `cuda-toolkit` is required for building `tiny-cuda-nn`. -For CUDA 11.7: - -```bash -pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 --extra-index-url https://download.pytorch.org/whl/cu117 - -conda install -c "nvidia/label/cuda-11.7.1" cuda-toolkit -pip install ninja git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch -``` - For CUDA 11.8: ```bash -pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 +pip install torch==2.1.2+cu118 torchvision==0.16.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 conda install -c "nvidia/label/cuda-11.8.0" cuda-toolkit pip install ninja git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch diff --git a/docs/quickstart/installation.md b/docs/quickstart/installation.md index 55cee8f1ef4..c3b756682b1 100644 --- a/docs/quickstart/installation.md +++ b/docs/quickstart/installation.md @@ -42,12 +42,12 @@ pip uninstall torch torchvision functorch tinycudann ``` ::::{tab-set} -:::{tab-item} Torch 2.0.1 with CUDA 11.8 +:::{tab-item} Torch 2.1.2 with CUDA 11.8 (recommended) -Install PyTorch 2.0.1 with CUDA 11.8: +Install PyTorch 2.1.2 with CUDA 11.8: ```bash -pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 +pip install torch==2.1.2+cu118 torchvision==0.16.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 ``` To build the necessary CUDA extensions, `cuda-toolkit` is also required. We