pytorch安裝使用

pytorch官網：https://pytorch.org/get-started/locally/

conda添加清華源：

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 --set show_channel_urls yes

添加第三方源：

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/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

創建環境：

conda create -n  torch_env python==3.7
conda activate torch_env 

下載（版本官網有對應，去掉-c pytorch）

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0

測試：

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

測試gpu:

import torch
torch.cuda.is_available()