반응형
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
반응형
'머신러닝,딥러닝' 카테고리의 다른 글
핸즈온 머신러닝 교재 코드 올라간 사이트 (0) | 2023.04.24 |
---|---|
conda에 gcc 설치하기 명령어 (0) | 2023.02.15 |
kaggle tensorflow version 오류 -> 코랩가서 하기 (0) | 2023.02.03 |
kaggle ~ RuntimeError: DGL requires tensorflow>=2.2.0 for the official DLPack support. 에러와 해결방법 (0) | 2023.02.01 |
[추천시스템 논문] CML : Collaborative Metric Learning 논문 리뷰 (0) | 2023.01.13 |
[GNN 관련 논문] TransE: Translating Embeddings for Modeling Multi-relational Data 논문 리뷰 (0) | 2023.01.11 |
유클리드 거리 vs 제곱 유클리드 거리 (squared euclidean distance or SED) (0) | 2023.01.11 |
dgl과 gpu를 사용할 때 에러가 날 경우 (0) | 2023.01.03 |