什么情况自己建设网站,dede替换网站模板,温州市网站制作多少钱,漯河网站建设xknt如题#xff0c;在本地跑xai的grok-1的时候遇见的问题。
首先你的cuda应该是安装好的#xff0c;也就是bash下nvidia-smi可以显示、python下torch.cuda.is_available()返回True。
在执行
import jaxjax.local_device_count()的时候报错这个。
在/usr/local/cuda/extras/C…如题在本地跑xai的grok-1的时候遇见的问题。
首先你的cuda应该是安装好的也就是bash下nvidia-smi可以显示、python下torch.cuda.is_available()返回True。
在执行
import jaxjax.local_device_count()的时候报错这个。
在/usr/local/cuda/extras/CUPTI/lib64应该是包含so文件的这时候在~/.bashrc加上export LD_LIBRARY_PATH$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64然后source ~/.bashrc就行了。
import jax
jax.devices()显示应该是[cuda(id0), cuda(id1), cuda(id2), cuda(id3), cuda(id4), cuda(id5), cuda(id6), cuda(id7)]这样的有几个卡就显示几个。
如果没有显示那要看你是不是安装的jax的gpu版本参考JAX: 库安装和GPU使用解决不能识别gpu问题
