为何Anaconda环境中PyTorch已安装成功但CUDA不可用(系统Python可用)
问题描述
在Windows 10系统中安装CUDA与PyTorch后,创建Anaconda环境并安装了相同组件。系统Python环境下PyTorch可正常识别GPU且CUDA可用,但Anaconda虚拟环境中CUDA无法使用。安装Numba排查时发现找不到cudadevrt.lib;多次重装CUDA、Python、Conda及相关依赖,甚至在新虚拟服务器测试仍出现相同问题。
系统Python中执行torch.cuda相关命令结果正常:
>>> import torch >>> torch.cuda.is_available() True >>> torch.cuda.current_device() 0 >>> torch.cuda.get_device_name() 'NVIDIA GeForce GTX 1080' >>>
Numba检测输出信息如下:
硬件信息 机器架构 : AMD64 CPU名称 : znver1 CPU核心数 : 16 可访问CPU核心数 : 16 可访问CPU核心列表 : 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 CFS运行时CPU限制 : 无 CPU特性 : 64bit adx aes avx avx2 bmi bmi2 clflushopt clzero cmov cx16 cx8 f16c fma fsgsbase fxsr lzcnt mmx movbe mwaitx pclmul popcnt prfchw rdrnd rdseed sahf sha sse sse2 sse3 sse4.1 sse4.2 sse4a ssse3 xsave xsavec xsaveopt xsaves 总内存(MB) : 32720 可用内存(MB) : 18803 操作系统信息 平台名称 : Windows-10-10.0.18362-SP0 平台版本 : 10 系统名称 : Windows 系统版本 : 10.0.18362 系统详细版本 : 10 10.0.18362 SP0 多处理器免费版 Libc版本 : 未知 Python信息 Python编译器 : MSC v.1916 64位(AMD64) Python实现 : CPython Python版本 : 3.10.9 Python区域设置 : en_IN.cp1252 Numba工具链版本 Numba版本 : 0.56.4 llvmlite版本 : 0.39.1 LLVM信息 LLVM版本 : 11.1.0 CUDA信息 CUDA设备已初始化 : 是 CUDA驱动版本 : 11.7 CUDA运行时版本 : 11.7 CUDA NVIDIA绑定可用 : 否 CUDA NVIDIA绑定正在使用 : 否 CUDA检测输出: 找到1个CUDA设备 ID 0: b'NVIDIA GeForce GTX 1080' [支持] 计算能力: 6.1 PCI设备ID: 0 PCI总线ID: 12 UUID: GPU-446ba761-7d06-bde9-66d7-10b96089d724 看门狗: 已启用 计算模式: WDDM FP32/FP64性能比: 32 总结: 1/1个设备受支持 CUDA库测试输出: 从CUDA_HOME查找nvvm 文件名: nvvm64_40_0.dll 尝试打开库... 成功 从CUDA_HOME查找cudart 文件名: cudart64_110.dll 尝试打开库... 成功 从CUDA_HOME查找cudadevrt 文件名: cudadevrt.lib 错误: 找不到cudadevrt: cudadevrt.lib 未找到 从CUDA_HOME查找libdevice 尝试打开库... 成功 NumPy信息 NumPy版本 : 1.23.5 NumPy支持的SIMD特性 : ('MMX', 'SSE', 'SSE2', 'SSE3', 'SSSE3', 'SSE41', 'POPCNT', 'SSE42', 'AVX', 'F16C', 'FMA3', 'AVX2') NumPy支持的SIMD调度 : ('SSSE3', 'SSE41', 'POPCNT', 'SSE42', 'AVX', 'F16C', 'FMA3', 'AVX2', 'AVX512F', 'AVX512CD', 'AVX512_SKX', 'AVX512_CLX', 'AVX512_CNL') NumPy支持的SIMD基线 : ('SSE', 'SSE2', 'SSE3') 检测到NumPy AVX512_SKX支持 : 否 SVML信息 SVML状态(config.USING_SVML) : 是 SVML库已加载 : 是 llvmlite使用SVML补丁版LLVM : 是 SVML可运行 : 是 线程层信息 TBB线程层可用 : 是 +-->TBB导入成功 OpenMP线程层可用 : 是 +-->厂商: MS Workqueue线程层可用 : 是 +-->Workqueue导入成功 Numba环境变量信息 未找到任何相关环境变量 Conda信息 Conda构建版本 : 3.24.0 Conda环境版本 : 23.3.1 Conda平台 : win-64 Conda Python版本 : 3.10.9.final.0 Conda根目录可写 : 是
内容的提问来源于stack exchange,提问作者VisuaL HippocracY
相关产品推荐
相关产品推荐

