linux 安装cuda10.1的 pytorch

安装 CUDA 10.1

1. 下载 CUDA 10.1 安装包

在 NVIDIA 官网上下载 CUDA 10.1 安装包。选择适合你的操作系统和显卡型号的版本。下载地址为 https://developer.nvidia.com/cuda-10.1-download-archive-base

2. 安装 CUDA 10.1

解压下载好的文件，进入目录，运行以下命令安装：

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

根据提示进行安装，选择合适的选项即可。

3. 配置环境变量

安装完成后，需要配置环境变量。在 ~/.bashrc 文件中添加以下内容：

export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH

然后运行以下命令使其生效：

$ source ~/.bashrc

安装 PyTorch

1. 安装依赖

PyTorch 依赖于以下软件包：

- Python 3.5 或以上版本
- NumPy
- setuptools
- wheel
- pyyaml
- cffi
- ninja
- mkl
- mkl-include

使用以下命令安装这些依赖：

$ sudo apt-get install python3-dev python3-pip
$ pip3 install numpy setuptools wheel pyyaml cffi ninja

2. 安装 PyTorch

使用以下命令安装 PyTorch：

$ 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。

至此，你已经成功安装了 CUDA 10.1 和 PyTorch，并且可以在 Python 中使用它们了。