ubuntu安装:Caffe, Torch, Cudnn,CUDA
时间: 2024-12-19 12:25:08 浏览: 17
Ubuntu上安装Caffe、Torch、CuDNN以及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`
阅读全文