Ubuntu系统上使用CUDA、CUDNN、PyTorch

文章结构概述

• • • • 1、配置cuda10.1环境
    • 2、部署cudnn 10.1组件
    • 3、完成pytorch框架的安装
    • • 3.1 libtorch模块
      • 3.2 pytorch核心库

1、CUDA 10.1安装指南

可通过下方链接获取cuda10.1的toolkits
cuda 10.1下载链接
依照示意图进行选择

按照Base Installer中所列的四条安装步骤进行操作，其中第二条指令cuda-repo-需要进行调整。在完成第一条指令的执行后，从终端日志的最后一行内容中直接复制并粘贴至命令行中。CUDA Toolkit将协助