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()