1、安装wsl2
具体过程网上一搜一把,这里就先略过了,只有wsl2能用哈
2、wsl里装docker,及相关配置
装docker
装完后有个小问题,wsl里的systemctl没法用,所以要编辑 /etc/wsl.conf 文件,添加:
wsl --shutdown 停止后再启动。
然后是某些总所周知的原因国内没法用dockerhub比较烦,这里推荐 这个教程 来解决。
3、给wsl装上 NVIDIA Container Toolkit,并让wsl能识别显卡
只用CPU来跑ollama的话可以跳过这一步,不过CPU真的太慢了。
装 NVIDIA Container Toolkit:
然后 nvidia-smi 看一下(提示没nvidia-smi命令的话 apt-get install装一下),这时候极大可能看到的列表为空。
查阅n卡官网的说明我们得知,还需要去 https://www.nvidia.com/Download/index.aspx 这个地址下载对应你显卡版本的GeForce Game Ready 驱动程序。
下载安装,重启电脑,再敲nvidia-smi命令,看到如下信息就是wsl识别到显卡了:
4、docker里跑ollama
运行如下命令启动ollama,只用GPU的话把--gpus=all这个去掉就行:
如下命令拉取想要的模型
然后运行如下命令就可以对话了
或者调web接口:
可以看到GPU用起来了:
虽然笔者的GPU是老掉牙的MX150,但性能明显比CPU模式要好很多,CPU跑这个问题要3分钟左右才有响应,但是GPU10多秒就开始有流输出了。但奇怪的是流输出开始后,GPU的使用率立马又掉下去了,不知道这个是bug还是feature还是笔者哪里没弄对,后面再研究一下。