ShravanHN commited on
Commit
121c9da
·
1 Parent(s): f581f01

torch was incompatible when upgraded

Browse files
Files changed (2) hide show
  1. app.py +4 -0
  2. requirements.txt +3 -1
app.py CHANGED
@@ -8,6 +8,10 @@ import logging
8
  import spaces
9
  from functools import lru_cache
10
 
 
 
 
 
11
  # Set up logging
12
  logging.basicConfig(level=logging.INFO)
13
  logger = logging.getLogger(__name__)
 
8
  import spaces
9
  from functools import lru_cache
10
 
11
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
12
+ # True
13
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
14
+
15
  # Set up logging
16
  logging.basicConfig(level=logging.INFO)
17
  logger = logging.getLogger(__name__)
requirements.txt CHANGED
@@ -1,4 +1,6 @@
1
  accelerate
2
  transformers==4.42.4
3
  SentencePiece
4
- bitsandbytes
 
 
 
1
  accelerate
2
  transformers==4.42.4
3
  SentencePiece
4
+ bitsandbytes
5
+ --extra-index-url https://download.pytorch.org/whl/cu113
6
+ torch