您的位置:首页 > 娱乐 > 明星 > 河南郑州疫情发布会_怎么做网页动图_互联网+营销策略怎么写_seo推广优化多少钱

河南郑州疫情发布会_怎么做网页动图_互联网+营销策略怎么写_seo推广优化多少钱

2025/2/25 18:38:09 来源:https://blog.csdn.net/weixin_44988127/article/details/145540778  浏览:    关键词:河南郑州疫情发布会_怎么做网页动图_互联网+营销策略怎么写_seo推广优化多少钱
河南郑州疫情发布会_怎么做网页动图_互联网+营销策略怎么写_seo推广优化多少钱

一、安装CudaToolkit

使用控制台命令查看当前显卡驱动中的cuda版本

nvidia-smi

二、下载CudaToolkit

注意:CudaToolkit版本要低于上面的显卡Cuda版本。
Cuda各版本下载地址:CUDA Toolkit Archive | NVIDIA Developer
下载好之后,一路下一步,一直默认即可。
如果安装到了自定义的位置,记得加环境变量。这里我安装到了D盘,下面是我添加的环境变量。

三、验证是否安装成功

nvcc --version

安装成功,如下图:

四、设置Ollama环境变量

Ollama安装好后,为了让推理跑在GPU上,可以按照如下步骤 设置环境变量:
在“系统变量”中,点击“新建”按钮。
添加以下环境变量:
变量名:OLLAMA_GPU_LAYER
变量值:cuda
如果需要指定特定的 GPU,可以添加以下环境变量:
变量名:CUDA_VISIBLE_DEVICES
变量值:GPU的UUID(按编号有时找不到,所以使用UUID)
在控制台输入nvidia-smi -L,即可查看GPU的UUID

五、查看ollama是否使用了gpu

通过观察PROCESSOR就能看到,大模型是用的cpu还是gpu,还是混合的。

ollama ps

经验证:8G的显卡,跑8b的模型可以100%用GPU,非常流畅。跑14b的cpu和gpu基本4/6开,可以观察SIZE模型运行需要的大小。

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com