Pytorch的安装
前言
为了省电赛的备赛,我开始安装Pytorch,计划为后面搭建神经网络为虹膜识别做个基础。
几个名词:
anaconda
是某个Python的开源版本,本质上是python以及各个依赖包的集合。安装pytorch之前要先安装ANANCONDA!
conda是什么?
conda是包及其依赖项和环境的管理工具。
conda包和环境管理器包含 于Anaconda的所有版本当中。
大家开手机的5G热点给电脑下载pytorch,不要怕流量贵,时间的成本更高。真的是经验之谈!!
WARNING conda.gateways.disk.delete:unlink_or_rename_to_trash
我觉得👆的报错肯定是因为多次下载之后肯定有什么内部的缓存文件没有被删除干净,还保留在虚拟环境之内。
一开始的时候重复创建了下载了很多次pytorch的包都没有成功,所以不断地关闭命令行打开命令行重新来过。重复多几次,居然发现每次进行都会有一大堆的报错。说“.ddl need to be remove muanuliy"。后来索性不解决,直接把现有的虚拟环境给移除。
conda info --envs // 查看当前情况下的环境
conda remove -n xxx --all // 清楚所有环境中的xxx这个虚拟环境
再重新创建虚拟环境:
conda create -n Pytorch python==3.9.0 // 创建虚拟环境名称为Pytorch的虚拟环境,使用Python3.9.0的python版本。
conda activate Pytorch // 激活虚拟环境
// 可能创建的过程中会有报错:
Solving environment: failed with repodata from current_repodata.json, will retry with next repodata source.
// 但是只要出现了Solving environment: done 就不会有问题 。
去官方网站获取拉取命令:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch // 记得配合着自己的电脑的情况选择
// 我的电脑是GTX1650
如果官方网站下载速度很慢的话,可以考虑换一下镜像源,具体可以参考这篇博客里的方法:
【Pytorch】pytorch安装的时候下载不动 - 知乎 (zhihu.com)
然后要注意的是。一定要在conda新建立的虚拟环境之下才能执行上面的更换镜像源的代码。
跑完条之后,有个操作可以验证一下,命令行输入python,然后import torch。试试看有无报错,没有的话应该是安装好了。
True表示torch可以被我们该电脑的GPU所使用。
祝大家的环境搭建顺利!!!