ubuntu16+nvidia驱动+cuda+an +pytorch安装

一、ubuntu16.04

        通过镜像安装

二、nvidia驱动

       1、 命令行输入 ubuntu-drivers devices 会出现可以安装以及推荐的驱动，后面带recommended就是推荐的版本

       2、安装 驱动 

  

sudo apt-get install nvidia-465

        3、不过我安装的是384的版本，安装好后输入 nvidia-smi，就会显示驱动状态

三、cuda安装，cuda版本要对应pytorch的版本，我安装的是11.1的cuda版本

        进入下载连接，点击版本，选择对应的选项 CUDA Toolkit Archive | NVIDIA Developer

        

        我选择的是11.1的版本 CUDA Toolkit 11.1 Update 1 Downloads | NVIDIA Developer

        然后最下面是选择安装方式，我现在的是deb的本地安装方式

                按照他给出的命令输入就好了,最后他强行要更新我的显卡驱动到465，行吧行吧，更就更了吧，更新了要重启驱动才能正常

        

四、pytorch安装

        打开下载网址 Start Locally | PyTorch 选择自己的配置,我是安装了anaconda,没有安装可以选择pip安装，好像pytorch有要求python3.6及以上，具体自己测试

        然后输入下面的命令

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

         

此时输入y