YALEIYALEIDAZE的博客

bug记录：streamlit mainpage与pytorch无法设置和成功使用显卡，tf可以。

streamlit的环境是py3.8，去噪任务环境是py3.7，torch版本是1.13.1。安装torch的时候不仅可以指定torch的版本还可以指定torch自带的cuda的版本。根据以往的经验，环境问题嘛肯定是，我又改了torch的版本，然后又出来cuda的错误了。做了一个Streamlit的小网站，想把一个torch的去噪任务集成上去。我3.7+torch1131不跑的挺好的嘛cuda我又没动过，怎么回事？他说我没有，那这是什么，我以前怎么用的显卡？会不会是这个cuda和这个torch的问题。