2021年4月14日
cuda pytorch torchvision版本对应问题
摘要： 最近开始学习nlp，安装fairseq，要求pytorch版本>=1.5，而服务器的cuda版本为10.0和9.2的，直接通过conda安装会自动安装对应cuda10.2的pytorch，影响使用。现记录解决这一问题的过程。 #1. cuda，pytorch，torchvision的对应关系如下图： 阅读全文
posted @ 2021-04-14 17:39 彭源 阅读(8008) 评论(0) 推荐(0) 编辑