pytorch版本的选择
pytorch版本和cuda版本一般有一定的适配关系
一般来说,先查看nvidia-smi在cuda driver api的版本,决定所需要下载的cuda的版本(不得高于driver版本),然后再看与cuda版本适配的torch版本
如,我的电脑中,Nvidai控制面版显示的cuda(driver api)版本为11.6, 因此可以支持cuda(runtime api, 即cudatoolkit)11.6及以下的版本,然后去pytorch官网查看cuda和torch版本对应关系
Previous PyTorch Versions | PyTorch
如下,一般官网命令都搭配好了版本
本人选择的是cu116版本对应的pytorch
直接使用pip install torch==1.12.1+cu116 .............命令下载,并且在torch、torchvison后都要+cu116, 最后下载的是gpu版本
本人的cuda driver API是11.6 ,一年前从官网装过cuda(cudatoolkit), nvcc -V返回的是11.2, 而pytorch适配的是cuda11.6,因此推测, 只要pytorch适配的cuda版本低于等于驱动版本,就都可以用, 即便cudatoolkit是11.2(当然也有可能是pip命令下载时整合了cudatoolkit11.6?) 最终输入torch.cuda.is_available()返回的也是True,训练代码目前也没有问题
07-11
5546
01-11
1214