2501_92387800的博客

原创 新人菜鸟Linux系统安装gpu的torch的踩坑实录

主包刚开始没注意，直到检查torch.cuda.is_available一直说false，主包才发现事情烦的严重性，最后检查到pytorch是鸟清华源的CPU版（就是上图的第一行），本来打算conda unstall的，可是实在太慢，主包直接强势重装。首先CUDA要对应合适的torch版本，其他文章已经说过了，这次说最容易踩坑的，这边装的方法是直接在终端 离线pip install下载已经下好的.whl文件，然后conda torch list。现在运行检测程序就是这样。希望能给大家一点帮助。