Spaces:
No application file
No application file
File size: 731 Bytes
3883c60 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from setup_tools.magicinstaller.requirement import Requirement
class Torch(Requirement):
def is_right_version(self):
ver = self.get_package_version('torch')
if ver:
# Check if a CUDA version is installed
return ver.startswith('2') and ('+cu' in ver if self.is_windows() else True)
return False
def is_installed(self):
return self.install_check('torch')
def install(self):
# if self.is_windows():
return self.install_pip('torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --extra-index-url https://download.pytorch.org/whl/cu118', 'PyTorch')
# else:
# return self.install_pip('torch torchvision torchaudio', 'PyTorch')
|