ubuntu安装gpu版pytorch,对应cuda12.4
时间: 2024-09-19 15:06:24 浏览: 32
要在Ubuntu系统上安装GPU版本的PyTorch,对应CUDA 12.4,你需要遵循以下步骤:
1. **更新系统**:
```sh
sudo apt-get update && sudo apt-get upgrade
```
2. **安装必要的依赖**:
```sh
sudo apt install -y software-properties-common curl git libffi-dev python3-dev build-essential cmake libssl-dev libcurl4-openssl-dev pkg-config
```
3. **添加CUDA PPA** (Personal Package Archive):
```
sudo add-apt-repository ppa:ubuntu-toolchain-r/test
```
然后更新软件源:
```sh
sudo apt-get update
```
4. **安装CUDA 12.4**:
找到CUDA 12.4的官方下载链接(注意选择适用于Ubuntu的版本),然后运行相应的.deb文件进行安装。例如:
```sh
sudo dpkg -ubuntu_<architecture>.deb
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu/<version>/<architecture>/x86_64/
sudo apt-get update
sudo apt-get install cuda=12.4
```
5. **安装cuDNN**:
```sh
export CUDA_HOME=/usr/local/cuda
wget https://developer.nvidia.com/rdp/cudnn-archive/v7.6.5.32-1_cuda10.2_caffe2-prod -O cuDNN-v7.6.5.32-1_cuda10.2-cuda12.4-linux-x64.tgz
tar xzf cuDNN-v7.6.5.32-1_cuda10.2-cuda12.4-linux-x64.tgz
sudo cp -r cuda/include/* /usr/local/include/
sudo cp -r cuda/lib64/* /usr/local/lib/
```
6. **安装Python 3.7+(如果未安装)**:
```sh
sudo apt-get install python3.7
```
7. **安装PyTorch**:
```sh
pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/torch_stable.html --extra-index-url https://download.pytorch.org/whl/cpu only
```
如果上述命令没有找到对应的CUDA 12.4版本,可以尝试手动指定:
```sh
pip3 install torch==1.9.0+cu124 torchvision==0.10.0+cu124 torchaudio==0.9.0 -f https://download.pytorch.org/whl/cu124/torch_stable.html
```
8. **验证安装**:
```sh
python3 -c "import torch; print(torch.cuda.is_available())"
```