硬件介绍CPU:i7-9700F,内存为8G,GPU为GTX1660
最近在跑一个代码(用pytorch写的),代码通了,但是运行速度很慢,且运行时GPU占用率很低。我按照龙良曲的tensorflow入门里的教程,安装了CUDA及CUDNN,在命令窗口输入nvcc -V,输出的命令也正常,但是用torch.is_available()。输出却为False。有没有大佬会,跪谢
pytorch要安装带gpu的版本 https://pytorch.org/get-started/locally/
谢谢你的回复,我是在anaconda下创建了一个环境,然后直接在anaconda的环境下装的pytorch,如下图所示,直接装的是CPU版本的嘛
这个我不太清楚是不是,得看你安装得时候用得命令是不是带gpu的版本的命令 你看我发的哪个链接 里面有各个版本的命令 或者重新安装一下 选pip 然后在它给的命令后面加上清华源安装很快
好的,谢谢您,我今天就试一下您说的方法。
cuda最高支持10.1的,但是现在安装了10.2的,怎么办,急求,有没有详细步骤啊!!