반응형

CUDA 3

Ubuntu 20.04 CUDA 재설치

1. 서론 cuda를 설치하는 것이 은근히 오래 걸리고 어렵기 때문에 정리하고자 한다. 2. 2.1 CUDA 삭제 sudo apt-get --purge -y remove 'cuda*' sudo apt-get --purge -y remove 'nvidia*' sudo apt-get autoremove --purge cuda # cudnn remove cd /usr/local/ sudo rm -rf cuda* ~/.bashrc나 /etc/profile에 추가되어있는 CUDA 관련 설정도 제거해야 한다. export PATH=$PATH:/usr/local/cuda-11.0/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-{version}/lib64 e..

개발환경 구축 2022.06.14

XOR 문제 딥러닝 모델 구현 (2) [Pytorch]

1. 서론 지난 글 XOR 문제 딥러닝 모델 구현 (1) [Pytorch]에서 input이 3개인 모델을 구현하였는데 이번엔 필요한 인풋 2개만 사용하여 은닉층의 최소 개수를 살펴보고자 한다. XOR 문제 딥러닝 모델 구현 (1) [Pytorch] 1. 서론 학부시절 딥러닝 첫걸음 책으로 딥러닝을 처음 접했다. 단층 신경망으로 풀 수 없는 XOR 문제를 심층 신경망으로 해결하는 코드가 교재에 matlab으로 구현되어있는데 이것을 Python의 Pytorch jeo96.tistory.com 2. 본론 지난 글에서 사용하였던 코드에서 입력 노드와 은닉 노드의 개수만 수정하였으며 모델 모양은 그림 1과 같다. 코드는 아래와 같으며 지난 코드에서 Linear의 숫자만 수정하였다. import torch devi..

Pytorch 2022.04.25

XOR 문제 딥러닝 모델 구현 (1) [Pytorch]

1. 서론 학부시절 딥러닝 첫걸음 책으로 딥러닝을 처음 접했다. 단층 신경망으로 풀 수 없는 XOR 문제를 심층 신경망으로 해결하는 코드가 교재에 matlab으로 구현되어있는데 이것을 Python의 Pytorch로 구현해 보았다. 2. 본론 input: [0, 0, 1], [0, 1, 1], [1, 0, 1], [1, 1, 1] output: [0], [1], [1], [0] 위와 같이 첫 번째와 두 번째 숫자가 같으면 0 다르면 1을 출력하는 모델을 구현하였다. 모델 모양은 그림 1과 같다. 3개의 입력을 받아 5개의 은닉 노드를 가지고 출력이 1개인 딥러닝 모델이다. 코드는 아래와 같다. import torch device = "cuda" if torch.cuda.is_available() else ..

Pytorch 2022.04.24
반응형