1.在pytorch官网下载之前,可以先在pytorch虚拟环境中添加清华镜像的源加快下载速度。
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
注意的是,pytorch给的下载链接如
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
里面的-c pytorch会导致1.38g的pytorch依然默认从官网下载,依然速度很慢,所以需要删除其中的-c pytorch。同理-c nvidia也会从官网下载。
2.pytorch下载完成后,输入python进入python中
3.输入import torch导入pytorch
4.输入print(torch.cuda.is_available())来检查是否配置完成,返回true即完成
5.输入exit()或者quit()来退出python