Ubuntu下安装Pytorch

安装CPU版本

https://www.cnblogs.com/wangqinze/p/13407610.html

完成后检验是否有误

    python
    import torch
    import torchvision

若没有报错则完成。

安装GPU版本

在配置好CUDA和cudnn的前提下,直接使用官网所给命令:

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

遇到以下报错时：
CondaHTTPError: HTTP 000 CONNECTION FAILED for url https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/linux-64