安装库报错:No CUDA runtime is found, using CUDA_HOME=‘/usr/local/cuda-11.3‘

1、报错内容

安装库时报错:

No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda-11.3'

2、检查

查看cuda版本和pytorch版本

python 进入python环境
import torch 
torch.__version__
torch.cuda.is_available()
nvidia-smi
nvcc -V

因此发现是由于该虚拟环境中CUDA与torch的版本不对应,发现在安装虚拟环境的environment.yml文件中只有指定cuda版本,而没有自动安装正确的torch版本:

channels:
  - pytorch
  - conda-forge
dependencies:
  - python
  - cudatoolkit=11.3

3、解决

重新安装torch,在 torch 官网中找到适配的版本,并安装

pip install torch1.12.1+cu113 torchvision0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

注意:尽量使用pip安装,因为conda安装后有时还是不会安装cuda对应版本

(Spatial-Mamba) ctuav_shixi@ecm-1b26:~/Spatial-Mamba-main/kernels/dwconv2d$ python3 setup.py install --user A module that was compiled using NumPy 1.x cannot be run in NumPy 2.0.1 as it may crash. To support both 1.x and 2.x versions of NumPy, modules must be compiled with NumPy 2.0. Some module may need to rebuild instead e.g. with &#39;pybind11>=2.12&#39;. If you are a user of the module, the easiest solution will be to downgrade to &#39;numpy<2&#39; or try to upgrade the affected module. We expect that some modules will need time to support NumPy 2. Traceback (most recent call last): File "/home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/setup.py", line 2, in <module> from torch.utils import cpp_extension File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/__init__.py", line 1382, in <module> from .functional import * # noqa: F403 File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/functional.py", line 7, in <module> import torch.nn.functional as F File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/nn/__init__.py", line 1, in <module> from .modules import * # noqa: F403 File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/nn/modules/__init__.py", line 35, in <module> from .transformer import TransformerEncoder, TransformerDecoder, \ File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/nn/modules/transformer.py", line 20, in <module> device: torch.device = torch.device(torch._C._get_default_device()), # torch.device(&#39;cpu&#39;), /home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/nn/modules/transformer.py:20: UserWarning: Failed to initialize NumPy: _ARRAY_API not found (Triggered internally at /opt/conda/conda-bld/pytorch_1695392067780/work/torch/csrc/utils/tensor_numpy.cpp:84.) device: torch.device = torch.device(torch._C._get_default_device()), # torch.device(&#39;cpu&#39;), running install /home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/cmd.py:90: SetuptoolsDeprecationWarning: setup.py install is deprecated. !! ******************************************************************************** Please avoid running ``setup.py`` directly. Instead, use pypa/build, pypa/installer or other standards-based tools. See https://blog.ganssle.io/articles/2021/10/setup-py-deprecated.html for details. ******************************************************************************** !! self.initialize_options() /home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/cmd.py:90: EasyInstallDeprecationWarning: easy_install command is deprecated. !! ******************************************************************************** Please avoid running ``setup.py`` and ``easy_install``. Instead, use pypa/build, pypa/installer or other standards-based tools. See https://github.com/pypa/setuptools/issues/917 for details. ******************************************************************************** !! self.initialize_options() running bdist_egg running egg_info writing dwconv2d.egg-info/PKG-INFO writing dependency_links to dwconv2d.egg-info/dependency_links.txt writing top-level names to dwconv2d.egg-info/top_level.txt reading manifest file &#39;dwconv2d.egg-info/SOURCES.txt&#39; writing manifest file &#39;dwconv2d.egg-info/SOURCES.txt&#39; installing library code to build/bdist.linux-x86_64/egg running install_lib running build_py copying Dwconv/dwconv_layer.py -> build/lib.linux-x86_64-cpython-310/Dwconv running build_ext /home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/utils/cpp_extension.py:424: UserWarning: There are no g++ version bounds defined for CUDA version 12.1 warnings.warn(f&#39;There are no {compiler_name} version bounds defined for CUDA version {cuda_str_version}&#39;) building &#39;dwconv2d&#39; extension Emitting ninja build file /home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/build/temp.linux-x86_64-cpython-310/build.ninja... Compiling objects... Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N) [1/1] /usr/local/cuda/bin/nvcc -I/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/include -I/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/include/TH -I/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/include/THC -I/usr/local/cuda/include -I/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/include/python3.10 -c -c /home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/depthwise_fwd/launch.cu -o /home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/build/temp.linux-x86_64-cpython-310/depthwise_fwd/launch.o -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr --compiler-options &#39;&#39;"&#39;"&#39;-fPIC&#39;"&#39;"&#39;&#39; -DTORCH_API_INCLUDE_EXTENSION_H &#39;-DPYBIND11_COMPILER_TYPE="_gcc"&#39; &#39;-DPYBIND11_STDLIB="_libstdcpp"&#39; &#39;-DPYBIND11_BUILD_ABI="_cxxabi1011"&#39; -DTORCH_EXTENSION_NAME=dwconv2d -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_80,code=compute_80 -gencode=arch=compute_80,code=sm_80 -std=c++17 FAILED: /home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/build/temp.linux-x86_64-cpython-310/depthwise_fwd/launch.o /usr/local/cuda/bin/nvcc -I/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/include -I/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/include/TH -I/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/include/THC -I/usr/local/cuda/include -I/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/include/python3.10 -c -c /home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/depthwise_fwd/launch.cu -o /home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/build/temp.linux-x86_64-cpython-310/depthwise_fwd/launch.o -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr --compiler-options &#39;&#39;"&#39;"&#39;-fPIC&#39;"&#39;"&#39;&#39; -DTORCH_API_INCLUDE_EXTENSION_H &#39;-DPYBIND11_COMPILER_TYPE="_gcc"&#39; &#39;-DPYBIND11_STDLIB="_libstdcpp"&#39; &#39;-DPYBIND11_BUILD_ABI="_cxxabi1011"&#39; -DTORCH_EXTENSION_NAME=dwconv2d -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_80,code=compute_80 -gencode=arch=compute_80,code=sm_80 -std=c++17 /home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/depthwise_fwd/launch.cu(29): error: namespace "at" has no member "NoGradGuard" at::NoGradGuard no_grad; ^ /home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/depthwise_fwd/launch.cu(63): error: namespace "at" has no member "NoGradGuard" at::NoGradGuard no_grad; ^ /home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/depthwise_fwd/launch.cu(69): error: namespace "at" has no member "NoGradGuard" at::NoGradGuard no_grad; ^ 3 errors detected in the compilation of "/home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/depthwise_fwd/launch.cu". ninja: build stopped: subcommand failed. Traceback (most recent call last): File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2100, in _run_ninja_build subprocess.run( File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/subprocess.py", line 526, in run raise CalledProcessError(retcode, process.args, subprocess.CalledProcessError: Command &#39;[&#39;ninja&#39;, &#39;-v&#39;]&#39; returned non-zero exit status 1. The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/ctuav_shixi/Spatial-Mamba-main/kernels/dwconv2d/setup.py", line 5, in <module> setup( File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/__init__.py", line 117, in setup return distutils.core.setup(**attrs) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/core.py", line 186, in setup return run_commands(dist) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/core.py", line 202, in run_commands dist.run_commands() File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/dist.py", line 1002, in run_commands self.run_command(cmd) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/dist.py", line 1104, in run_command super().run_command(command) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/dist.py", line 1021, in run_command cmd_obj.run() File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/command/install.py", line 109, in run self.do_egg_install() File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/command/install.py", line 167, in do_egg_install self.run_command(&#39;bdist_egg&#39;) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/cmd.py", line 357, in run_command self.distribution.run_command(command) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/dist.py", line 1104, in run_command super().run_command(command) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/dist.py", line 1021, in run_command cmd_obj.run() File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/command/bdist_egg.py", line 177, in run cmd = self.call_command(&#39;install_lib&#39;, warn_dir=False) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/command/bdist_egg.py", line 163, in call_command self.run_command(cmdname) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/cmd.py", line 357, in run_command self.distribution.run_command(command) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/dist.py", line 1104, in run_command super().run_command(command) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/dist.py", line 1021, in run_command cmd_obj.run() File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/command/install_lib.py", line 19, in run self.build() File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/command/install_lib.py", line 113, in build self.run_command(&#39;build_ext&#39;) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/cmd.py", line 357, in run_command self.distribution.run_command(command) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/dist.py", line 1104, in run_command super().run_command(command) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/dist.py", line 1021, in run_command cmd_obj.run() File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/command/build_ext.py", line 99, in run _build_ext.run(self) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/command/build_ext.py", line 368, in run self.build_extensions() File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 873, in build_extensions build_ext.build_extensions(self) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/command/build_ext.py", line 484, in build_extensions self._build_extensions_serial() File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/command/build_ext.py", line 510, in _build_extensions_serial self.build_extension(ext) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/command/build_ext.py", line 264, in build_extension _build_ext.build_extension(self, ext) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/Cython/Distutils/build_ext.py", line 136, in build_extension super().build_extension(ext) File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/setuptools/_distutils/command/build_ext.py", line 565, in build_extension objects = self.compiler.compile( File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 686, in unix_wrap_ninja_compile _write_ninja_file_and_compile_objects( File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1774, in _write_ninja_file_and_compile_objects _run_ninja_build( File "/home/ctuav_shixi/anaconda3/envs/Spatial-Mamba/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2116, in _run_ninja_build raise RuntimeError(message) from e RuntimeError: Error compiling objects for extension安装包的时候报错应该怎么办
05-14
2025-05-13 21:17:23,497 SSD INFO: Using 1 GPUs 2025-05-13 21:17:23,498 SSD INFO: Namespace(config_file=&#39;configs/vgg_ssd300_voc0712.yaml&#39;, local_rank=0, log_step=10, save_step=2500, eval_step=2500, use_tensorboard=True, skip_test=False, opts=[], distributed=False, num_gpus=1) 2025-05-13 21:17:23,498 SSD INFO: Loaded configuration file configs/vgg_ssd300_voc0712.yaml 2025-05-13 21:17:23,498 SSD INFO: MODEL: NUM_CLASSES: 5 INPUT: IMAGE_SIZE: 300 DATASETS: TRAIN: ("voc_2007_trainval", "voc_2012_trainval") TEST: ("voc_2007_test", ) SOLVER: MAX_ITER: 120000 LR_STEPS: [80000, 100000] GAMMA: 0.1 BATCH_SIZE: 32 LR: 1e-3 OUTPUT_DIR: &#39;outputs/vgg_ssd300_voc0712&#39; 2025-05-13 21:17:23,499 SSD INFO: Running with config: DATASETS: TEST: (&#39;voc_2007_test&#39;,) TRAIN: (&#39;voc_2007_trainval&#39;, &#39;voc_2012_trainval&#39;) DATA_LOADER: NUM_WORKERS: 8 PIN_MEMORY: True INPUT: IMAGE_SIZE: 300 PIXEL_MEAN: [123, 117, 104] MODEL: BACKBONE: NAME: vgg OUT_CHANNELS: (512, 1024, 512, 256, 256, 256) PRETRAINED: True BOX_HEAD: NAME: SSDBoxHead PREDICTOR: SSDBoxPredictor CENTER_VARIANCE: 0.1 DEVICE: cuda META_ARCHITECTURE: SSDDetector NEG_POS_RATIO: 3 NUM_CLASSES: 5 PRIORS: ASPECT_RATIOS: [[2], [2, 3], [2, 3], [2, 3], [2], [2]] BOXES_PER_LOCATION: [4, 6, 6, 6, 4, 4] CLIP: True FEATURE_MAPS: [38, 19, 10, 5, 3, 1] MAX_SIZES: [60, 111, 162, 213, 264, 315] MIN_SIZES: [30, 60, 111, 162, 213, 264] STRIDES: [8, 16, 32, 64, 100, 300] SIZE_VARIANCE: 0.2 THRESHOLD: 0.5 OUTPUT_DIR: outputs/vgg_ssd300_voc0712 SOLVER: BATCH_SIZE: 32 GAMMA: 0.1 LR: 0.001 LR_STEPS: [80000, 100000] MAX_ITER: 120000 MOMENTUM: 0.9 WARMUP_FACTOR: 0.3333333333333333 WARMUP_ITERS: 500 WEIGHT_DECAY: 0.0005 TEST: BATCH_SIZE: 10 CONFIDENCE_THRESHOLD: 0.01 MAX_PER_CLASS: -1 MAX_PER_IMAGE: 100 NMS_THRESHOLD: 0.45 D:\SSDcode\SSD-master\ssd\utils\model_zoo.py:62: FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle modu le implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#u ntrusted-models for more details). In a future release, the default value for `weights_only` will be flipped to `True`. This limits the functions that could be executed during unpi ckling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via `torch.serialization.add_safe_globals`. We rec ommend you start setting `weights_only=True` for any use case where you don&#39;t have full control of the loaded file. Please open an issue on GitHub for any issues related to this experimental feature. return torch.load(cached_file, map_location=map_location) 2025-05-13 21:17:24,109 SSD.trainer INFO: No checkpoint found. 2025-05-13 21:17:24,154 SSD.trainer INFO: Start training ... Traceback (most recent call last): File "D:\SSDcode\SSD-master\train.py", line 114, in <module> main() File "D:\SSDcode\SSD-master\train.py", line 105, in main model = train(cfg, args) File "D:\SSDcode\SSD-master\train.py", line 44, in train model = do_train(cfg, model, train_loader, optimizer, scheduler, checkpointer, device, arguments, args) File "D:\SSDcode\SSD-master\ssd\engine\trainer.py", line 91, in do_train loss.backward() File "D:\wulianwang\an\an1\envs\ssd\lib\site-packages\torch\_tensor.py", line 581, in backward torch.autograd.backward( File "D:\wulianwang\an\an1\envs\ssd\lib\site-packages\torch\autograd\__init__.py", line 347, in backward _engine_run_backward( File "D:\wulianwang\an\an1\envs\ssd\lib\site-packages\torch\autograd\graph.py", line 825, in _engine_run_backward return Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR_HOST_ALLOCATION_FAILED
05-14
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值