Solving environment: failed with initial frozen solve. Retrying with flexible solve.NVIDIA驱动更新装torch
1.问题描述:笔者做项目需要跑一个卷积模型,根据作者描述torch版本2.4.0还要搭配cuda12.4。
随即进入torch官网(Previous PyTorch Versions | PyTorch)找到对应安装指令。
具体指令为:
# CUDA 12.4
conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 pytorch-cuda=12.4 -c pytorch -c nvidia
输入指令后发现一直停滞不动
经过搜索引擎后,问题得以解决。
2.尝试方法:
2.1.尝试一
网上大部分原因被归结于网络不太行,所以推荐换源。这里给出清华源:
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
部分可能会用到的指令,查看镜像源:
conda config --show channels
重置镜像源:
conda config --remove-key channels
2.2.尝试二
其实安装cuda要先查看自己笔记本电脑能承受的最大版本,在终端中输入nvidia-smi
(使用anaconda安装pytorch环境参考:手把手教你使用anaconda安装pytorch环境(适合新手)-CSDN博客)
右上角cuda版本告诉你你最大只能装11.2的版本(轻薄本是这样的,有没有大佬捐个游戏本给俺玩玩儿,哭qaq|模型自己跑不了了),再次去torch官网查看发现只能支持pytorch==1.12.1,cuda==10.2等版本的了。
以管理员身份(可选)打开Anaconda Prompt (anaconde)终端,进入指定虚拟环境,并输入指令(我的就是conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch)没一会儿就安装好了。由于我已经成功安装了,这里只显示下成功截图。
2.3.说明:其他的尝试,诸如anaconda更新,anaconda缓存清理等没有尝试。
3.参考资料:
(不是镜像问题,是命令问题!)Anaconda安装Pytorch报错:PackagesNotFoundError - 知乎
补更新:
1.显卡驱动更新
笔者后来了解到,虽然是轻薄本,但是仅仅是显存大小小(如:仅有2GB),版本是可以更新到相对较新的程度的(如:CUDA Version: 13.0)。
打开任务管理器,找到性能->GPU(NVIDIA),查看自己显卡的具体型号
NVIDIA显卡驱动更新地址:下载 NVIDIA 官方驱动 | NVIDIA
根据自己上述任务管理器的型号进行选择->查找,随机页面跳转,点击 “查看”
页面再次跳转
点击“下载”,于是开始下载安装包,如下图
双击启动安装,这个默认是要下载到C盘的,不需要改动Extraction path,直接点击“OK”
点击“同意并继续”
根据自己需求完成安装选项的选取,博主是轻薄本,所以选择“精简(E)(推荐)”->“下一步”
进入“准备安装”,等待完成即可。
等待一段时间之后,显示安装完成
2.检查驱动更新是否成功
再次进入终端cmd,输入
nvidia-smi
CUDA版本从CUDA Version:11.2变为CUDA Version:13.0
这样就可以去Pytorch官网下载对应的torch版本训练自己的模型啦。
给一张我安装成功的截图(我安装了torch2.8.0版本的,不是指定的2.4.0,能用就行,嘻嘻),这个是之前已经安装好的,展示下结果,进入虚拟环境,pip list。
更多推荐
所有评论(0)