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。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐