Ubuntu下安装pytorch相关问题

Ubuntu下安装pytorch相关问题

Ubuntu安装问题

u盘制作启动盘，安装无脑下一步。安装完卡界面，bios改uefi

安装可以进不了可能是没有引导，就repair引导，try ubuntu without install

1. sudo add-apt-repository ppa:yannubuntu/boot-repair && sudo apt-get update
2. sudo apt-get install -y boot-repair && boot-repair
3. 选择recommended repair

cuda cudnn

1. 安装好显卡驱动

2. 官网选择好cuda安装，

   sudo sh cuda版本.run
   

   不安装驱动，剩下都勾选

3. 设置环境变量 修改bashrc

   export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda版本/lib64
   export PATH=$PATH:/usr/local/cuda版本/bin
   export CUDA_HOME=$CUDA_HOME:/usr/local/cuda版本
   

   应用修改的bashrc

   source ~/.bashrc
   

   检查成功与否:

   nvcc --version
   

4. 下载对应cudnn

   sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
    
   sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
    
   sudo chmod a+r /usr/local/cuda/include/cudnn.h
    
   sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
   

   验证:

   cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
   

pytorch

pytorch官网选择对应的命令行

验证：

import torch
torch.cuda.is_available()