linux 安装pytorch cu101

安装 PyTorch with CUDA 10.1

1. 安装 CUDA 10.1

在开始安装 PyTorch with CUDA 10.1 之前，需要先安装 CUDA 10.1。可以在 NVIDIA 官网上下载 CUDA 10.1 安装包。选择适合你的操作系统和显卡型号的版本。下载地址为 https://developer.nvidia.com/cuda-10.1-download-archive-base

下载完成后，解压文件并运行以下命令安装：

$ sudo sh cuda_10.1.243_418.87.00_linux.run

2. 安装 cuDNN

cuDNN 是一个 CUDA 深度神经网络库，可以加速深度学习的计算。在安装 PyTorch with CUDA 10.1 之前，需要先安装 cuDNN。可以在 NVIDIA 官网上下载 cuDNN 安装包。选择适合你的操作系统和显卡型号的版本。下载地址为 https://developer.nvidia.com/rdp/cudnn-download

下载完成后，解压文件并运行以下命令安装：

$ sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb

3. 安装 PyTorch

使用以下命令安装 PyTorch with CUDA 10.1：

$ pip3 install torch torchvision

以上命令将会安装最新版本的 PyTorch 和 torchvision。

如果你需要安装特定版本的 PyTorch，可以使用以下命令：

$ pip3 install torch==1.4.0 torchvision==0.5.0

这将会安装 PyTorch 1.4.0 和 torchvision 0.5.0。

至此，你已经成功安装了 PyTorch with CUDA 10.1。