pip 快速安装pytorch torchvision
1.先在命令窗口查看自己的cuda对应版本,如果有独立显卡,可使用cuda加速器处理,输入指令为nvidia-smi,我的版本为cuda11.4
2.可在cuda官网找到对应版本,参考其他专门安装cuda和cudnn的方式,安装好cuda和cudnn,我的已安装了cuda11.4。
3.安装pytorch,先进去pytorch官网:https://pytorch.org/get-started/locally/#windows-anaconda,选择自己电脑对应的pytorch版本,我的是cuda11.4,但官网最新为cuda11.3,所以我选了最新的,自己电脑cuda高一些没关系。我选了pip安装,对应最底部出现了安装指令:pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
4.当然可以安装这条指令安装,但是下载速度会很慢,时间久了容易报时间错误,因此,最快最稳定的方法就是,点击后面的链接,自己手动下载所需要的版本,https://download.pytorch.org/whl/cu113,点开链接如下图所示
5.可以点击自己想安装的模块,安装pytorch时,先安装torch,再安装torchvision。
**安装torch:**点击torch,可看到如下界面,我选择了我电脑对应版本,python3.7,当然没有cuda的,可选择cpu版本,(最好建立一个虚拟环境,conda creat pytorch python==3.7)
4.下载到本地后,命令窗口cd到所在文件夹,使用pip install torch-1.10.1+cu113-cp37-cp37m-win_amd64.whl,可实现pytorch的安装。
5.torchvision的安装过程类似,选择对应的版本即可,需要特别注意的是,使用pip install 安装时,会自动下载cpu的torch,卸载掉自己的cuda版本,解决办法是指令变为:pip install --no-deps 你的torchvision名称。但是这样安装后,导入会报错,缺一些库,比如numpy,pillow等,对应安装上就好。
6,安装库的时候,最好后面加镜像源,速度会更快
例如:pip install pillow -i https://pypi.tuna.tsinghua.edu.cn/simple/
我也试过conda install 安装方法,总会有问题,亲测这种方法既快又不会出错,祝大家好运!!!