파이토치 쿠다 버전 관련 문제를 격고 있습니다

Question

python-3.6.9pytorch-1.4.0 py3.6_cpu_0 [cpuonly] pytorchtorchvision-0.5.0 py36_cpu [cpuonly] pytorchcuda-10.2cudatoolkit-10.1.243 import torch torch.cuda.is_available()False 제 컴퓨터 환경이 이러합니다. File &#34;scripts/train_rpn_3d.py&#34;, line 190, in <module>main(sys.argv[1:])File &#34;scripts/train_rpn_3d.py&#34;, line 51, in maininit_torch(conf.rng_seed, conf.cuda_seed)File &#34;/kinematic3d/lib/core.py&#34;, line 675, in init_torchtorch.set_default_tensor_type('torch.cuda.FloatTensor')File &#34;/opt/conda/lib/python3.6/site-packages/torch/__init__.py&#34;, line 155, in set_default_tensor_type_C._set_default_tensor_type(t)TypeError: type torch.cuda.FloatTensor not available. Torch not compiled with CUDA enabled. 맨 아랫줄에 &#34;TypeError: type torch.cuda.FloatTensor not available. Torch not compiled with CUDA enabled.&#34; 오류를 해결하고 싶습니다..

Myoungdo Park · Answer

cpu only 버전이 설치된 듯 하네요. pytorch 는 cpu or cuda 버전 별로 설치명령어가 다릅니다.pytorch install 페이지 참고하셔서 원하시는 옵션의 설치 명령어로 시도해 보세요.

파이토치 쿠다 버전 관련 문제를 격고 있습니다

1개의 답변

#torch 목록