Day66_Ai(4)_Pm / 딥러닝전 설치

gpu 버전 = 실수

nvidia 

cuda 설치

cmd => nvidia-smi로 확인

내 버전에 맞는걸 쓰면 된다.

 

 

 

---

 

 

---

 

 

 

만약 gpu가 없으면 쿠다는 다운받지말고 cpu 버전을 찾아서 해라

 

 

---

 

test

import torch

print('pytorch version : ', torch.__version__)
#pytorch version :  2.6.0+cu126
print('CUDA available :', torch.cuda.is_available())
#CUDA available : True

if torch.cuda.is_available():
    print('CUDA version : ', torch.version.cuda)
    print('cuDNN version : ', torch.backends.cudnn.version())
    print('GPU name : ', torch.cuda.get_device_name(0))
    print('GPU count : ', torch.cuda.device_count())

    a = torch.rand(3,3).cuda()
    b = torch.rand(3,3).cuda()
    c = torch.matmul(a, b)
    print('\n GPU matrix multiply success')
    print(c)