diff --git a/requirements.txt b/requirements.txt index 754accf..ca60aef 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ # Model torch>=1.13.1 torchvision>=0.14.1 -cupy-cuda11x>=11.0.0 +cupy-cuda11x==11.0.0 # Monitor/Metric tensorboardX>=2.6