1 pytorch安装(CPU版本)
pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
2 torchvision、torchaudio、torchtext安装:解决ModuleNotFoundError: No module named ‘torchvision‘问题
(conda list查明没有torchvision的前提下)
pip install torchvision -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
3 pytorch gpu版本安装
3.1 检查本机GPU版本
此电脑→右键“管理”
我的Graphics630是集显,GeForce GTX 1050是独显,分不清可以自己去查查,如果只有集显下面的就没法继续了。
3.2 安装NVIDIA驱动及控制面板
https://www.nvidia.cn/Download/index.aspx?lang=cn
进入官网根据自己的配置进行选择
我选的是这个,因为是家族所以不会跟自己的配置一模一样
点击搜索,看看支持列表有没有自己的型号,框出来的是我的配置
下载后安装没什么好讲的
3.3 CUDA版本
nvidia控制面板->帮助->系统信息->组件
如果显示
win+R,输入services.msc
把这个启动起来(但是这个界面最好别关,后面可能还要再打开)
这里就是CDUA的版本
3.4 下载对应版本的CUDA
https://developer.nvidia.com/cuda-toolkit-archive
3.5 CUDA安装
正常来说就是一直点下一步,但是如果最后报错nvidia backend 安装失败,还记不记我前边说别关的services.msc那个界面,找到并停止所有NVIDIA 开头的服务;
并打开任务管理器杀掉所有的nvidia进程;
然后重试,我这次就没有问题了
4 pytorch GPU版本
要装torch torchversion torchaudio三样,他们的版本之间,还有这三者和Python的版本都有严格的对应关系
可以参照这个
https://blog.csdn.net/shiwanghualuo/article/details/122860521
关于安装源,我试过官网的那个版本,cpu的时候还行,到了GPU版本的时候会一直卡在Solving environments这里
清华源也不行,据说是被封锁了
我用的基本上是这个方法
https://blog.csdn.net/i__om/article/details/128773034
在此网站下载whl文件
https://download.pytorch.org/whl/torch_stable.html
直接用人家博主的图做解释
按照这个逻辑找所需的torch torchversion torchaudio,但是注意这里的文件不全,先找好了再下,不然重新再来一次2、3个G下的挺慢的
5 pytorch GPU安装
把这三个文件放在prompt的默认路径下,一般是c:/usrer/username
然后
pip install
安装torch torchversion torchaudio的顺序依次安装,要写完整的文件名
例:
pip install torch-1.10.0+cu113-cp36-cp36m-win_amd64.whl
6 确认安装是否成功
输入python后,输入下列几行代码,如果能正常执行则说明安装成功
import torch
print(torch.__version__)
print(torch.cuda.is_available())
6.1 “OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.”异常
删除此环境对应路径下的libiomp5md.dll
我的环境存储在d:/anaconda/save/envs下
名称叫做face
可以做参考
(就离谱,装个包我能写2400+字数的记录,都是泪啊)