pytorch安装
直接pip/conda
pip install torch==xxx(版本号)
conda install pytorch-xxx(版本号)
以上两种是直接安装pytorch,后期若有cuda的报错,使用以下方法
投机取巧法
1.官方渠道
官网链接: https://pytorch.org/get-started/previous-versions/.
进入选择与cudn、python、torchvision、cudatoolkit相符合版本
比如需要cuda10.0的,pytorch为1.1.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch
如无意外,这样安装肯定会很慢,直接换源或许能行。
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/ma