2022年7月28日
PyTorch和TensorFlow使用的CUDA的差别
摘要： 最近遇到一个很有意思的问题，使用nvidia-smi能够查看CUDA版本，使用nvcc无法查看CUDA版本，找不到本机CUDA的安装路径，PyTorch-GPU可以正常使用，TensorFlow-GPU无法正常使用。 现将解决这个问题过程中查找的资料以及思考记录如下： CUDA的API包括：runt 阅读全文
posted @ 2022-07-28 18:28 云上飘啊飘 阅读(258) 评论(0) 推荐(0)