(2条消息) Pytorch-gpu版安装教程【注意:无需提前安装cuda和cudnn】_山风wind的博客-CSDN博客_pytorch一定要cuda吗
返回true了,说明已经可以用GPU跑代码了,不需要再安装cuda和cudnn了。当你执行安装命令的时候,对应版本的这两个东西,已经打包安装了。你可以切换了虚拟环境中,然后输入conda list命令查看,应该能看到,安装好的东西既包括torch_gpu,也包括cuda和cudnn。
conda应该会把pytorch依赖的cudnn自动处理好
更新一下:
李沐老师说的两者的区别