SakuraD commited on
Commit
c71518c
1 Parent(s): 0686ccc

update packages

Browse files
Files changed (2) hide show
  1. app.py +5 -1
  2. requirements.txt +4 -2
app.py CHANGED
@@ -1,5 +1,10 @@
1
  import os
2
  # import spaces
 
 
 
 
 
3
 
4
  # install packages for mamba
5
  def install():
@@ -10,7 +15,6 @@ def install():
10
 
11
  install()
12
 
13
- import torch
14
  import torch.nn as nn
15
  import numpy as np
16
  import torch.nn.functional as F
 
1
  import os
2
  # import spaces
3
+ import torch
4
+
5
+ os.system("nvidia-smi")
6
+ print("TORCH_CUDA", torch.cuda.is_available())
7
+
8
 
9
  # install packages for mamba
10
  def install():
 
15
 
16
  install()
17
 
 
18
  import torch.nn as nn
19
  import numpy as np
20
  import torch.nn.functional as F
requirements.txt CHANGED
@@ -8,5 +8,7 @@ setuptools
8
  timm
9
  transformers
10
  wheel
11
- torch==2.1.1+cu118 --find-links https://download.pytorch.org/whl/cu118
12
- torchvision==0.16.1+cu118 --find-links https://download.pytorch.org/whl/cu118
 
 
 
8
  timm
9
  transformers
10
  wheel
11
+ torch
12
+ torchvision
13
+ # torch==2.1.1+cu118 --find-links https://download.pytorch.org/whl/cu118
14
+ # torchvision==0.16.1+cu118 --find-links https://download.pytorch.org/whl/cu118