- 博客(1)
- 收藏
- 关注
原创 解决pytorch找不到gpu的一个方法(针对conda虚拟环境不在C盘 )
第一次安装pytorch后很顺利地能使用GPU,卸载后第二次安装pytorch却死活用不了GPU。检查显卡驱动、cuda、pytorch版本全都对应的,用的官网源没错下cpu版本。意识到第一次虚拟环境安装在默认C盘路径,第二次C盘红了就把环境装在了E盘,所以路径设置可能有问题。然后 torch.cuda.is_available() 就为true了。
2024-08-05 16:12:41 254
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人