安装cuda后遇到的问题(Ubuntu)

1.Could not load library libcudnn_cnn_infer.so.8. Error: libcuda.so: cannot open shared object file: No such file or directory.Please make sure libcudnn_cnn_infer.so.8 is in your library path!

        未添加环境变量。执行下面语句,添加。

 export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}

  export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64/stubs/:/usr/local/cuda-11.8/lib64:/usr/local/cuda-11.8/cudnn/lib:$LD_LIBRARY_PATH

2. UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 34: CUDA driver is a stub library (Triggered internally at  /opt/conda/conda-bld/pytorch_1659484809535/work/c10/cuda/CUDAFunctions.cpp:109.)return torch._C._cuda_getDeviceCount() > 0

        重新安装版本匹配的pytorch-cuda。

        用下面的代码测试。

import torch

# 检查是否有可用的GPU

if torch.cuda.is_available():

    device_count = torch.cuda.device_count()

    print(f"当前系统中可用的GPU数量为:{device_count}")

else:

    print("未检测到可用的GPU")

         解决:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值