ubuntu18之pytorch

pytoch官網

pytorch-gpu版本安裝

#官方提供的安裝命令，運行此命令後發現安裝的是pytorch cpu版本
#conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

#經過測試，使用以下命令可以完成安裝
#conda install pytorch-gpu torchvision -c pytorch
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

python
import torch
print(torch.cuda.is_available())     #測試gpu是否可用，如果可用，即安裝正確

安裝包下載速度慢，參考如下連接

清華源:下載安裝包

手動下載自己需要的對應pytorch版本安裝包，然後進行本地安裝
有的conda或pipy源太慢，conda install xxx或者pip install xxx下載會中斷連接導致壓縮包下載不全，本地的安裝包沒法完全安裝

#pip 安裝本地包
pip install   D:\XXX.whl

#conda 安裝本地包
conda install --use-local   ~/Downloads/pytorch-1.3.1-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2
conda install pytorch-gpu torchvision -c pytorch #繼續安裝其它依賴包

vim ~/.condarc
auto_activate_base: false
channels:
   - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
   - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
   - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
   - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
   - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
   - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
   - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
   - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
 show_channel_urls: true

tensorboard（推薦）

問題：pytorch ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed
解決方法：由於沒有安裝tensorboard引起的，安裝tensorboard即可

conda install tensorboard
tensorboard --logdir log_path   #顯示命令

python如下調用

from torch.utils.tensorboard import SummaryWriter

tensorboardX（不推薦使用）

conda install tensorboardX
tensorboard --logdir log_path    #顯示命令

python如下調用

from tensorboardX import SummaryWriter