ubuntu16.04pytorch0.4+ctc 구성
853 단어 인공지능
conda create -n pytorch python=3.6
2,pytorch0을 설치합니다.4
conda install pytorch=0.4.0 cuda80 -c pytorch
3. 컴파일ctc
git clone https://github.com/SeanNaren/warp-ctc.git
cd warp-ctc
mkdir build; cd build
cmake ..
make
컴파일 과정에서 문제가 생기지 않지만,
cd pytorch_binding
python setup.py install
이 프로세스에 문제가 발생할 수 있습니다.
/home/sobhe/repo/warp-ctc/pytorch_binding/src/binding.cpp:105:61: error: too few arguments to function ‘cudaError_t THCudaMalloc(THCState*, void**, size_t)
파일 수정 "pytorch binding/src/binding.cpp"1 at 92 line int probssize = THCudaTensor_size(state, probs, 2); 2 at l05 lines void* gpu_workspace; THCudaMalloc(state, &gpu_workspace, gpu_size_bytes);
다시 실행:
cd pytorch_binding
python setup.py install
축하합니다, 성공했습니다.
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
SPRESENSE x Neural Network Console을 만져 보자 제1회 ~SPRESENS의 소개와 전체의 구성~이 기사는 20일째 기사입니다. 전회는 Katsuaki Takagi씨의 「 」이었습니다. Sony가 발매한 보드 「SPRESENSE」와 「Neural Network Console」을 맞추어 사용해 봅니다. 연재 기사...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.