Pytorch使用
1.安装
本文章所依赖的基础软件:Anaconda
条件可以的话,首先以官方进行安装:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
在国内,最为推荐使用以下的方式进行安装:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=10.1