File tree Expand file tree Collapse file tree 1 file changed +14
-5
lines changed Expand file tree Collapse file tree 1 file changed +14
-5
lines changed Original file line number Diff line number Diff 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
20201 . 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
You can’t perform that action at this time.
0 commit comments