下载完pytorch之后想验证一下cuda好不好使,在测试的时候发现一个warning
python中报错如下
我下载的pytorch版本比较低,numpy太高,所以需要手动给numpy降级
pip install numpy\<2
降级后再进到python验证cuda就没有warning和报错了,顺利的运行了。
下载完pytorch之后想验证一下cuda好不好使,在测试的时候发现一个warning
python中报错如下
我下载的pytorch版本比较低,numpy太高,所以需要手动给numpy降级
pip install numpy\<2
降级后再进到python验证cuda就没有warning和报错了,顺利的运行了。