Sadjad Alikhani commited on
Commit
983540c
1 Parent(s): 15a15cf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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=11.8 -c pytorch -c nvidia
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
- scenario_idxs = np.array([3])
 
 
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)