我正在使用的计算机与Titan XP一起运行,并运行Ubuntu 18.10。我不是所有者,所以不确定之前的配置方式。 cuda版本是9. *,最有可能是9.0。没有像/usr/local/cuda
这样的文件夹。尽管这听起来很奇怪(因为没有Cuda与18.10兼容),但以前它在Tensorflow和Pytorch上都运行良好。现在,当在python 2.7,cudatoolkit 9.2和cudnn 7.2.1中运行tensorflow-gpu v1.12.0时(以前运行良好,没有任何更改),它报告:
ImportError: libcublas.so.9.0: cannot open shared object file: No such file of directory
但是,当我用pytorch 0.4.1,cudatoolkit 9.0和cudnn 7.6(将它们显示在pycharm中)将conda env更改为python 3.6时。有:
torch.cuda.is_available() # True
这表明GPU正在Pytorch代码中运行。另外,我已经通过nvidia-smi
检查了GPU RAM,当Pytorch运行时,RAM被占用了。
运行时,虽然没有像/usr/local/cuda/
这样的Cuda文件夹:
nvcc - V
有:
Cuda compilation tools,release 9.1,V9.1.85
有人可以给我一些有关这些奇怪事情如何发生的提示吗?我应该怎么做才能使我的tensorflow-gpu工作?我完全迷惑了Orz。