安装了GPU版本的torch却不能用?(已解决来看看)
将以下七个文件夹全部复制到对应的环境的目录下(即上面一个截图中envs-->unet_implementation-->lib-->site-packages),并把环境中对应的torch全家桶文件夹删除。希望能帮到同样问题的人。已经安装了GPU版本的torch,但是使用torch.cuda.is_available()来验证的时候得到的结果还是一个大大的false。但是我在pycharm中查看解释器中的环境:此时的torch版本又是另一个版本。③这样对比终端的库和pycharm的库一眼发现问题;
原创
2023-10-18 19:39:54 ·
1492 阅读 ·
2 评论