我正在尝试在 Google Colab 上运行 stylegan2,但所有文件都在我的云端硬盘上,并避免使用来自 stylegan2 github 的 !git clone。 这是我在特定单元格上的代码:
%tensorflow_version 1.x
%cd /content/drive/My Drive/stylegan2-master/
!nvcc test_nvcc.cu -o test_nvcc -run
print('Tensorflow version: {}'.format(tf.__version__) )
!nvidia-smi -L
print('GPU Identified at: {}'.format(tf.test.gpu_device_name()))
结果:/内容/驱动器/我的驱动器/stylegan2-master CPU 打招呼。 cudaErrorNoDevice:未检测到支持 CUDA 的设备 张量流版本:1.15.2
NVIDIA-SMI 失败,因为它无法与 NVIDIA 驱动程序通信。确保已安装并运行最新的 NVIDIA 驱动程序。
GPU 识别于:为什么我无法获取 GPU ? 我是这个领域的新手,所以我可能会错过一些非常简单的东西,但仍然无法在互联网上找到答案。
您可以通过单击
Edit > Notebook settings
并选择 GPU 作为硬件加速器来轻松完成此操作。应该是这样。
请使用此代码:
!nvidia-smi 2>&1 | grep -i“cuda”
!nvidia-smi
它帮助我解决错误。