wangjin2000 commited on
Commit
8c93f94
·
verified ·
1 Parent(s): ee16ecb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -2,7 +2,7 @@
2
  # $ conda create --name <env> --file <this file>
3
  # platform: linux-64
4
 
5
- pytorch==2.0.1
6
  pytorch-cuda==11.8
7
  accelerate==0.27.2
8
  scikit-learn==1.3.0
 
2
  # $ conda create --name <env> --file <this file>
3
  # platform: linux-64
4
 
5
+ pytorch==2.0.1py3.8_cuda11.8_cudnn8.7.0_0
6
  pytorch-cuda==11.8
7
  accelerate==0.27.2
8
  scikit-learn==1.3.0