不知道为什么同一个环境,.py代码能跑,但是jupyter的环境报了这个错误
感觉这个环境也挺奇怪的
首先,在命令行能够转成cuda tensor。
然后命令行输出的python版本和安装环境时制定的python版本不一致
佛了最后还是重装了环境:
conda create -n diffusion_demo python=3.10
pip install ipykernel
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
# CUDA 11.7
pip install numpy==1.* # 脚本还不支持numpy>2,我佛了