Initial commit: project structure, scripts and documentation
This commit is contained in:
@@ -0,0 +1,13 @@
|
||||
import torch
|
||||
|
||||
def check_cuda_availability():
|
||||
if torch.cuda.is_available():
|
||||
print("CUDA доступна!")
|
||||
print(f"Количество доступных GPU: {torch.cuda.device_count()}")
|
||||
for i in range(torch.cuda.device_count()):
|
||||
print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
|
||||
else:
|
||||
print("CUDA недоступна.")
|
||||
|
||||
if __name__ == "__main__":
|
||||
check_cuda_availability()
|
||||
Reference in New Issue
Block a user