1. 打开Anaconda Prompt,输入如下命令启动虚拟环境

      1. conda activate jupyter
    2. 安装PyTorch 和 torchvision

    直接使用conda install下载超级超级慢,还安装失败了,所以改用清华源。依次输入如下4条命令

    1. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    2. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    3. conda config --set show_channel_urls yes
    4. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

    打开Pytorch官网,根据你的电脑配置,获取相应的命令,但是要去掉最后的“-c pytorch”,否则还是官网下载。
    image.png

    1. conda install pytorch torchvision cudatoolkit=10.1
    1. 安装完毕,验证安装是否成功

    打开jupyter notebook,新建python文件,选择jupyter虚拟环境
    运行demo如下,代表安装成功。
    image.png