cuda版本为11.7 ubuntu20.04安装pytorch-gpu
时间: 2024-06-08 07:07:10 浏览: 21
你可以按照以下步骤在 Ubuntu 20.04 上安装 PyTorch GPU 版本:
1. 安装 CUDA 和 cuDNN
在 Ubuntu 20.04 上安装 CUDA 和 cuDNN 的步骤如下:
1.1 安装 CUDA
首先,你需要在 NVIDIA 官网上下载适合你的 CUDA 版本。CUDA 11.7 可以从以下链接下载:https://developer.nvidia.com/cuda-downloads。
下载完成后,你可以使用以下命令将其安装到系统中:
```bash
$ sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.109-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-7-local/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get install cuda
```
1.2 安装 cuDNN
cuDNN 库是用于深度学习加速的 CUDA 拓展库。你可以从 NVIDIA 官网下载适合你的 cuDNN 版本。cuDNN 8.2.4 可以从以下链接下载:https://developer.nvidia.com/rdp/cudnn-download。
下载完成后,你可以使用以下命令将其解压到 `/usr/local/cuda` 目录下:
```bash
$ tar -xzvf cudnn-11.7-linux-x64-v8.2.4.15.tgz
$ sudo cp -r cuda/include/* /usr/local/cuda/include/
$ sudo cp -r cuda/lib64/* /usr/local/cuda/lib64/
$ sudo ldconfig
```
2. 创建虚拟环境并安装 PyTorch
为了避免与系统中已有的 Python 库冲突,我们可以使用 Python 虚拟环境。可以使用以下命令创建虚拟环境:
```bash
$ python3 -m venv venv
$ source venv/bin/activate
```
然后,你可以使用以下命令安装 PyTorch:
```bash
$ pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
这将安装适合 CUDA 11.1 的 PyTorch 版本。
安装完成后,你可以使用以下命令验证是否安装成功:
```bash
$ python
>>> import torch
>>> print(torch.cuda.is_available())
```
如果返回 `True`,则说明安装成功。
3. 安装其他依赖项
在使用 PyTorch 进行深度学习时,你可能还需要安装其他一些库。你可以使用以下命令安装常用的深度学习库:
```bash
$ pip install numpy matplotlib scikit-learn jupyterlab
```
以上是在 Ubuntu 20.04 上安装 PyTorch GPU 版本的步骤,希望能够帮助到你。