torch使用cudnn7

本文提供了一种通过Git克隆cuDNN库并使用LuaRocks进行构建的具体步骤。此外,还提到了与cuDNN版本相关的已知问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Ubuntu上安装Caffe、TorchCuDNN以及CUDA是一个常见的深度学习开发环境设置步骤。以下是基本的安装指南: 1. **更新系统并安装依赖**: ```sh sudo apt-get update && sudo apt-get upgrade sudo apt-get install software-properties-common python-software-properties ``` 2. **添加CUDA和GPU驱动的官方存储库**: 如果你有NVIDIA GPU,运行: ```sh sudo add-apt-repository ppa:nvidia/cuda ``` 然后输入 `wget -q -c http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/README.txt` (替换为你的Ubuntu版本) 获取更多详细信息。 3. **安装CUDA**: 根据README文件指示,安装适合你的GPU的CUDA版本。例如: ```sh sudo apt-get install cuda ``` 4. **安装CuDNN**: 需要在官网下载CuDNN库,并将其移动到CUDA的include和lib目录下。如果你使用的是CUDA 11.x,可以去https://developer.nvidia.com/cudnn 下载对应版本。 5. **安装Python的Caffe库**: 使用pip安装: ```sh pip install caffe ``` 或者从源码安装(更推荐): - 克隆Caffe仓库 `git clone https://github.com/BVLC/caffe.git` - 进入目录并安装 `cd caffe && make -j$(nproc)` - 安装Python绑定 `make pyproto && python setup.py build install` 6. **安装PyTorch**: 可能需要先安装一些依赖如`gcc`, `python-dev`, `numpy`, 然后通过pip获取最新稳定版: ```sh pip install torch torchvision ``` 7. **检查安装**: 检查安装是否成功,例如运行 `python -c "import torch"` 和 `python -m caffe`
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值