Files changed (1) hide show
  1. README.md +4 -5
README.md CHANGED
@@ -85,22 +85,21 @@ Follow these steps to replicate our Conda environment and install the necessary
85
  #### Create and Activate Conda Environment
86
 
87
  ```
88
- conda create --name smi-ted-env python=3.8.18
89
  conda activate smi-ted-env
90
  ```
91
 
92
  #### Install Packages with Conda
93
 
94
  ```
95
- conda install pytorch=1.13.1 cudatoolkit=11.4 -c pytorch
96
- conda install numpy=1.23.5 pandas=2.0.3
97
- conda install rdkit=2021.03.5 -c conda-forge
98
  ```
99
 
100
  #### Install Packages with Pip
101
 
102
  ```
103
- pip install transformers==4.6.0 pytorch-fast-transformers==0.4.0 torch-optimizer==0.3.0 datasets==1.6.2 scikit-learn==1.3.2 scipy==1.12.0 tqdm==4.66.1
 
104
  ```
105
 
106
  ## Pretraining
 
85
  #### Create and Activate Conda Environment
86
 
87
  ```
88
+ conda create --name smi-ted-env python=3.10
89
  conda activate smi-ted-env
90
  ```
91
 
92
  #### Install Packages with Conda
93
 
94
  ```
95
+ conda install pytorch=2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia
 
 
96
  ```
97
 
98
  #### Install Packages with Pip
99
 
100
  ```
101
+ pip install -r requirements.txt
102
+ pip install pytorch-fast-transformers
103
  ```
104
 
105
  ## Pretraining