pytorch cuda 버전 문제 인것 같습니다. 도와주세요

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.” 오류를 해결하고 싶습니다…

설치된 PyTorch를 지우고 여기서 CUDA로 컴파일된 버전을 찾아 재설치해보세요.