728x90
apex를 install하기 전 확인해야할 것은 /usr/local/에 있는 CUDA 버전과 cudatoolkit 버전과 동일해야한다는 것이다. (cudatoolkit이 없을 경우 10.2라고 생각하면 됨)
만약 CUDA 버전을 다른 버전으로 install 해야한다면 CUDA Install를 보고 진행하면 된다.
버전을 동일하게 맞춰주었다면 이후부터는 apex github에 나와있듯이 다음과 같이 입력하여 install하면 된다.
* pytorch 1.9.0까지 지원하는 듯 하다.
* NVIDIA Driver는 설치할 CUDA 버전 이상의 버전이 설치되어있어야한다.
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
728x90
'Temp' 카테고리의 다른 글
| pytorch에서 사용되는 cuda는 system에 설치된 cuda인가 (0) | 2022.08.05 |
|---|---|
| [Linux] 다른 버전의 CUDA Install (0) | 2022.08.01 |
| [Docker] Get Started (0) | 2022.07.30 |
| [OpenCV] Polygon Crop (0) | 2022.05.09 |
| [JSON] json.loads() -> Expecting value: line 1 column 2 (char 1) (0) | 2022.03.28 |