로그인 바로가기 하위 메뉴 바로가기 본문 바로가기

Hands on Deep Learning

임시 이미지 이규빈 교수
http://gist.edwith.org/hands-on-deep-learning-01/forum/29279
좋아요 85 수강생 985

좋은 강의 오픈하여 주셔 감사합니다. 

.....

os: windows 10 

그래픽카드: nVidia GeForce GTX 1050


nVidia 설치되어 있는데 

import torch
print(torch.__version__)
a = torch.tensor([[1., -1.], [1., -1.]])
print(a)
print(torch.cuda.is_available())
if torch.cuda.is_available() == True:
print(a.cuda())

위 예제 프로그램을 실행하면 

1.6.0
tensor([[ 1., -1.],
[ 1., -1.]])
False

결과가 나옵니다. 

원인을 알 수 있을까요?

그리고 계속에서 같은 결과(예제)를 나오면 torch-cpu를 설치해야 하나요 아니면 설치없이 계속해도 오류없이 진행할 수 있나요?