본문 바로가기

카테고리 없음

jetson nano pycuda install 방법

jetson nano jetpack version : 4.5.1

 

python3 -m pip install pycuda

하면 오류가 날 것이다.

 

pip3로 하지말고 왠만하면 본인이 사용하는 python3의 pip를 명시해주자.

 

일단 내 경우에는 

 

이런 방식으로 해결하였다.

 

1. BASHRC 수정

우선 쿠다 경로부터 보면,

직접 키보드 탭을 치면서 확인하면서 진행하자, 내 것 그대로 가져가지 말고 확인하는 습관을 가져야한다.

/usr/local/ 에서 탭을 치면

cuda 폴더와 cuda-설치한버젼 폴더가 뜰것이다.

없으면 cuda 설치가 안된 것이니, 설치부터 하자

 

만일 있다면
바로 가기 경로인 cuda말고 cuda-설치한버젼을 확인하면된다.

내 경우는 cuda-10.2였다.

 

그 다음 

 

vim ~/.bashrc

** vim 이든 vi, nano 암거나

 

bashrc에 들어가면, 맨 마지막 줄로 가서 밑에 두 줄을 입력한다. (본인 버젼에 맞게 할 것!!)

export PATH="/usr/local/cuda-10.2/bin:$PATH"

export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH"

 

이 두 줄을 적은 후 저장한다. :wq

 

터미널에서

sudo source ~/.bashrc

 

// 설정이 잘 되었는지 확인하자

nvcc --version

안된다면 위에서 다시 꼼꼼히 확인

 

 

2. PYCUDA Install

(2022.12.2 수정)

"

sudo python3 -m pip install pycuda 를 해보자. () 

여기 하지말고 바로 밑으로 진행하자.

bashrc까지 한 뒤에

"

 

wget https://files.pythonhosted.org/packages/5e/3f/5658c38579b41866ba21ee1b5020b8225cec86fe717e4b1c5c972de0a33c/pycuda-2019.1.2.tar.gz

tar xvf pycuda-2019.1.2.tar.gz

cd pycuda-2019.1.2

python3 configure.py --cuda-root=/usr/local/cuda-10.2

python3 setup.py install

 

 

그럼 설치가 될 것이다.