Research Notes

[Pytorch] CUDA와 Pytorch 버전 맞추기 본문

Programming Language/Python

[Pytorch] CUDA와 Pytorch 버전 맞추기

jiachoi 2022. 4. 14. 14:25

1.  Cuda 버전 확인 

- window cmd 창에서 아래 코드로 Cuda version 확인 

>> nvcc - V

- cuda 11.0 버전임

 

2. Pytorch와 CUDA 버전 맞추기 

- 아래 링크에서 CUDA 버전에 맞게 Pytorch 설치 

https://pytorch.org/get-started/previous-versions/

 

PyTorch

An open source machine learning framework that accelerates the path from research prototyping to production deployment.

pytorch.org

  

저작자표시

'Programming Language > Python' 카테고리의 다른 글

[Solved] IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number  (0) 2022.09.22
[Python] Virtual Environment Generation Reference  (0) 2022.08.04
[Solved] 'RuntimeError: CUDA out of memory.'  (0) 2022.04.27
[Solved] TypeError: an Integer is required (got type bytes)  (0) 2022.04.23
[Solved] OSError: [WinError 127] 지정된 프로시저를 찾을 수 없습니다  (0) 2022.04.14
'Programming Language/Python' Related Articles more