一、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

标签: none

评论已关闭

粤ICP备2025485561号-1