分类：CUDA版本

2021
03-17

解决PyTorch与CUDA版本不匹配的问题

1.CUDA驱动和CUDAToolkit对应版本表一：CUDA驱动及CUDAToolkit最高对应版本最新可查阅官方文档注：驱动是向下兼容的，其决定了可安装的CUDAToolkit的最高版本。2.CUDAToolkit版本及其可用PyTorch对应版本（参考官网）表二：CUDAToolkit版本及可用PyTorch对应关系CUDAToolkit版本可用PyTorch版本7.50.4.1，0.3.0，0.2.0，0.1.12-0.1.68.01.1.0，1.0.0，0.4.... 继续阅读 >

2020
10-08

pytorch 查看cuda 版本方式

由于pytorch的whl安装包名字都一样，所以我们很难区分到底是基于cuda的哪个版本。有一条指令可以查看importtorchprint(torch.version.cuda)补充知识：pytorch：网络定义参数的时候后面不能加".cuda()"pytorch定义网络__init__()的时候，参数不能加“cuda()",不然参数不包含在state_dict()中，比如下面这种写法是错误的self.W1=nn.Parameter(torch.FloatTensor(3,3),requires_grad=True).cuda()应该去掉".cuda()"self.W1=n... 继续阅读 >