当我运行“torch.rand(10).to(“cuda”)”时,我遇到“错误:Torch 未在启用 CUDA 的情况下编译”
GPU:Nvidia RTX 3080 Ti
$nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 525.85.05 Driver Version: 525.85.05 CUDA Version: 12.0 |
|-------------------------------+----------------------+----------------------+
Run Code Online (Sandbox Code Playgroud)
$nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85
Run Code Online (Sandbox Code Playgroud)
$conda list
cudatoolkit version : 11.0.221
$conda info
active environment : sw
active env location : /mnt/user2/.conda/envs/sw
shell level : 1
user config file : /mnt/user2/.condarc
populated config files :
conda version : 23.1.0
conda-build version : …
Run Code Online (Sandbox Code Playgroud)