harpreetsahota commited on
Commit
2c70e90
1 Parent(s): d2db8f9

install a version of PyTorch compatible with the built-in CUDA drivers

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,2 +1,3 @@
 
1
  torch
2
  transformers
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
  torch
3
  transformers