使用 PyTorch 测试 CUDA 是否安装成功
由于老唐之前都是使用同学装好的 GPU 服务器的,一直没有自己装过,今天自己装了一下,发现还挺复杂的。本文先记录一下如何使用 PyTorch 测试 CUDA 是否安装成功,测试 CUDA 是否安装成功还是挺简单的,由于一直使用 PyTorc […]
由于老唐之前都是使用同学装好的 GPU 服务器的,一直没有自己装过,今天自己装了一下,发现还挺复杂的。本文先记录一下如何使用 PyTorch 测试 CUDA 是否安装成功,测试 CUDA 是否安装成功还是挺简单的,由于一直使用 PyTorc […]