各位同学也许还在疑惑:
为什么当我print(torch.cuda.is_available())为True时,竟然还会发现当用到cuda库中的函数时无法调用或者在命令行写出nvcc -V时提示:command not found. 这是因为torch.cuda.is_available()为True并不意味着cuda已经安装成功,它只是代表你的系统装好了GPU驱动!!!
让我们来看看这是为什么:
在pytorch框架下想使用cuda,首先要明确的是你需要一个GPU,以及符合GPU型号的nvidia驱动,**再加上一个cuda工具包(一般称为cudatoolkit)**这些我想在看我的帖子之前你已经看了很多,这里点出来一个容易混淆的地方,有些帖子所说的cuda driver就是一个适用于GPU的nvidia driver!驱动只需要安装一次,尤其是对于那些容器而言,大部分已经预装了相应的nvidia driver,如果想使用cuda,最好还是去看看cudatoolkit是否已经安装,你说不知道是不是已经安装,这里有两个方法:
第一个:nvcc -V 如果安装好了会提示cuda的版本,如下图:
第二个:去寻找cuda相应的文件,cuda包含了一些函数库,cuda目录下面总有一系列lib文件,找到了cuda根据地的所在,可以说已经安装好了,至于为什么不能用那就可能是其他的问题。
所以当torch.cuda.is_available()为true时,有可能你的电脑上连cuda还没有装,在Linux系统下登陆网站利用wget命令下载下来吧,如果是Windows系统,也可以去官网直接下载。