cuda PyTorch

1. GPU对应的CUDA版本 nvidia-smi

 CUDA Version: 12.2   

GPU diver 大于cuda toolkit, pytorch 版本根据cuda toolkit

2. 查看nvcc的版本(即cuda toolkit 版本)

nvcc --version 


Cuda compilation tools, release 12.2, V12.2.91
Build cuda_12.2.r12.2/compiler.32965470_0
 

如未安装,下载对应的cuda版本安装 不要安装驱动

下载对应的cuda

cuda pytorch

cuda 下载Pytorch深度学习实战3-1:最新Windows/Ubuntu双系统Pytorch图文安装教程-CSDN博客

你可能感兴趣的:(pytorch,人工智能,python)