Anaconda中安装PyTorch

1.进入官网
https://pytorch.org/
选择对应的版本的命令
在这里插入图片描述
然后,打开anaconda Prompt,输入上述命令即可。
这种方法,如果网速不好,会导致安装失败。

2.上述方法,会由于网络等原因,安装失败。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cpuonly pytorch (把c去掉)

3.直接下载安装包进行安装

4.pip安装(或者pip3)
这里是python3.6版本
这个是有cuda
pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

这个是cpu
pip install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip install https://download.pytorch.org/whl/cpu/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

猜你喜欢

转载自blog.csdn.net/AugustMe/article/details/101600237
今日推荐