python-3.6.9
pytorch-1.4.0 py3.6_cpu_0 [cpuonly] pytorch
torchvision-0.5.0 py36_cpu [cpuonly] pytorch
cuda-10.2
cudatoolkit-10.1.243
import torch
torch.cuda.is_available()
False
제 컴퓨터 환경이 이러합니다.
File “scripts/train_rpn_3d.py”, line 190, in
main(sys.argv[1:])
File “scripts/train_rpn_3d.py”, line 51, in main
init_torch(conf.rng_seed, conf.cuda_seed)
File “/kinematic3d/lib/core.py”, line 675, in init_torch
torch.set_default_tensor_type(‘torch.cuda.FloatTensor’)
File “/opt/conda/lib/python3.6/site-packages/torch/init.py”, 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.
맨 아랫줄에 “TypeError: type torch.cuda.FloatTensor not available. Torch not compiled with CUDA enabled.” 오류를 해결하고 싶습니다…