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