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