anaconda安装不同版本的pytorch
由于一些工程原因需要安装不同版本的pytorch,这些pytorch依赖于不同版本的cuda。直接创建虚拟环境,然后使用官网的安装指令一般是没用的。1,先安装多个版本的cuda和cudnn,比如cuda9和cuda10。(注意,安装的时候显卡驱动,安装高版本的驱动,这里是cuda10对应的显卡驱动,nvidia-smi 出来的cuda版本是10.0,可以向下兼容。安装cuda9.0的时候显卡驱动就
·
由于一些工程原因需要安装不同版本的pytorch,这些pytorch依赖于不同版本的cuda。直接创建虚拟环境,然后使用官网的安装指令一般是没用的。
1,先安装多个版本的cuda和cudnn,比如cuda9和cuda10。(注意,安装的时候显卡驱动,安装高版本的驱动,这里是cuda10对应的显卡驱动,nvidia-smi 出来的cuda版本是10.0,可以向下兼容。安装cuda9.0的时候显卡驱动就可以不装了)。
2,假如要安装pytorch1.1(依赖于cuda9),先在~/.bashrc中链接到cuda9中,使用nvcc -V显示cuda9.0。然后创建虚拟环境,使用官网指令安装即可。
3,假如安装成功pytorch1.1,现在又要安装pytorch1.2(依赖于cuda10),同理,先在~/.bashrc中链接到cuda10.0中,使用nvcc -V显示cuda10.0。然后创建虚拟环境,使用官网指令安装即可。
最后,你想用哪个pytorch,激活对应的环境即可,不需要修改~/.bashrc。
over, good lucky!
更多推荐



所有评论(0)