【日常分享】服务器配置及安装cuda及PyTorch方法

前言

在服务器上配置完显卡及其驱动后，如果想跑深度学习代码，还需要安装cuda及深度学习框架，比如PyTorch，本文介绍其中的一些安装步骤。

安装cuda

借鉴：https://blog.csdn.net/sinat_36721621/article/details/115326307

第一步：在NVIDIA官网选择与系统对应的版本

按照上图输入两行命令即可完成cuda安装。

安装pytorch

在PyTorch官网 (https://pytorch.org/) 可以下载最新版本或之前的版本。

可以使用conda或者pip安装，注意pytorch和cuda版本对应即可。