Pytorch1.0rc + torchvision + fastai 安装

  • 首先,安装好anaconda环境,这个直接去官网(传送门)搞定就行
  • 然后,添加清华大学源(不然下载巨慢)
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
  • 然后,安装三连,基本和github上介绍一样,但是我发现安装pytorch是如果指定了channel -c 则不会使用清华源,然后cuda 依赖可以交给anconda自动解决,所以这个直接install就行
conda install pytorch-nightly
conda install -c fastai torchvision-nightly
conda install -c fastai fastai
  • 测试一下,安装是否成功,执行下面python脚本,能正常运行输出就行
import torch
import torchvision
import fastai.vision as fv
print("torch version: "+torch.__version__)
print("GPU is available: "+str(torch.cuda.is_available()))
print(fv.learner.AdamW)

你可能感兴趣的:(pytorch)