Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | 6 | |
7 | 8 | 9 | 10 | 11 | 12 | 13 |
14 | 15 | 16 | 17 | 18 | 19 | 20 |
21 | 22 | 23 | 24 | 25 | 26 | 27 |
28 | 29 | 30 |
Tags
- sigmoid
- XOR
- neural net
- overfitting
- machine learning
- Python
- Attention
- Transformer
- Set Transformer
- Self-loop attention
- sentence embedding
- pytorch
- Classificaion
- NLP
- GNN
- abstraction
- word2vec
- deep learning
- elif
- python practice
Archives
- Today
- Total
Research Notes
[Pytorch] CUDA와 Pytorch 버전 맞추기 본문
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