打开Anaconda Prompt,输入如下命令启动虚拟环境
conda activate jupyter
安装PyTorch 和 torchvision
直接使用conda install下载超级超级慢,还安装失败了,所以改用清华源。依次输入如下4条命令
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
打开Pytorch官网,根据你的电脑配置,获取相应的命令,但是要去掉最后的“-c pytorch”,否则还是官网下载。
conda install pytorch torchvision cudatoolkit=10.1
- 安装完毕,验证安装是否成功
打开jupyter notebook,新建python文件,选择jupyter虚拟环境
运行demo如下,代表安装成功。