Skip to content

Commit b89cde9

Browse files
authored
Update Tutorial.md
1 parent 3ce0829 commit b89cde9

File tree

1 file changed

+14
-5
lines changed

1 file changed

+14
-5
lines changed

Tutorial.md

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,22 +15,31 @@ If you encounter any issues during installation or usage, please refer to the [Q
1515
* Pytorch 1.7.1
1616
* NVIDIA GPU (GeForce RTX 3090) + CUDA (11.7)
1717

18-
#### Environment Configuration
18+
#### Environment Configuration
1919

2020
1. Create a virtual environment and install PyTorch.
2121

2222
```
2323
$ conda create -n udmt python=3.8
2424
$ conda activate udmt
25-
$ pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
2625
$ sudo apt-get install ninja-build
2726
$ sudo apt-get install libturbojpeg
2827
```
29-
30-
2. We made a installable pip release of UDMT [[pypi](https://pypi.org/project/udmt-pip/)]. You can install it by entering the following command:
28+
If your CUDA version is **12.x**, run:
29+
```
30+
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu121
31+
```
32+
If your CUDA version is **11.x**, run:
33+
```
34+
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
35+
```
36+
2. Install other dependencies.
3137
3238
```
33-
$ pip install udmt-pip
39+
$ conda activate udmt
40+
$ git clone https://github.com/cabooster/UDMT.git
41+
$ cd UDMT/
42+
$ pip install -r requirements.txt
3443
```
3544
3645
### 2. For Windows

0 commit comments

Comments
 (0)