Linux下Pytorch1.1.0(GPU)安装笔记

1.cuda, pytorch版本对应问题     设备上的GPU驱动版本决定了cuda版本,而cuda版本与cudatoolkit的版本一致, 而pytorch(GPU)的版本要与cudatoolkit的版本严格一致。      一开始没留意这个版本的对应关系,直接输入官网上给出的命令,踩了大坑!(官网上的cudatoolkit版本为9.2,而我设备上的cudatoolkit版本为9.0,不对应
相关文章
相关标签/搜索