Sadjad Alikhani
commited on
Commit
•
983540c
1
Parent(s):
15a15cf
Update README.md
Browse files
README.md
CHANGED
@@ -56,10 +56,8 @@ conda activate lwm_env
|
|
56 |
Install the necessary packages inside your new environment.
|
57 |
|
58 |
# Install CUDA-enabled Pytorch
|
59 |
-
You can find the script for installing the CUDA-enabled Pytorch based on your preferences here: https://pytorch.org/get-started/locally/
|
60 |
-
The following is an example script:
|
61 |
```bash
|
62 |
-
conda install pytorch torchvision torchaudio pytorch-cuda=
|
63 |
```
|
64 |
# Install other required packages from conda-forge
|
65 |
```bash
|
@@ -163,7 +161,9 @@ scenario_names = np.array(["city_18_denver",
|
|
163 |
"city_11_santaclara",
|
164 |
"city_7_sandiego"]
|
165 |
)
|
166 |
-
|
|
|
|
|
167 |
selected_scenario_names = scenario_names[scenario_idxs]
|
168 |
|
169 |
# Clone the requested scenario folders (this will clone every time)
|
|
|
56 |
Install the necessary packages inside your new environment.
|
57 |
|
58 |
# Install CUDA-enabled Pytorch
|
|
|
|
|
59 |
```bash
|
60 |
+
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
|
61 |
```
|
62 |
# Install other required packages from conda-forge
|
63 |
```bash
|
|
|
161 |
"city_11_santaclara",
|
162 |
"city_7_sandiego"]
|
163 |
)
|
164 |
+
|
165 |
+
# Choose the desired scenario or secanrios (if you need the combined scenarios as a larger and more diverse scenario.).
|
166 |
+
scenario_idxs = np.array([0,1,2,3,4,5,6])
|
167 |
selected_scenario_names = scenario_names[scenario_idxs]
|
168 |
|
169 |
# Clone the requested scenario folders (this will clone every time)
|