RTX2080显卡与pytorch兼容问题:cuda9.2

由于RTX2080显卡支持的主流cuda版本是10.0及以上,pytorch 0.4支持的最高版本cuda是9.2,而很多时候运行别人的代码需要用到pytorch 0.4(pytorch 1.0及以上版本不兼容),因此需要找到一个兼容的解决方案。

如果直接在anaconda里面装cuda9.0及以下版本,系统cuda用10.0,可以解决大部分问题,但是当需要用系统cuda编译的时候,系统cuda编译出来的库在conda环境下因为cuda版本不对没有办法运行,所以还需要与系统的cuda版本一致。

几经周折,发现RTX2080是支持cuda9.2的,pytorch也勉强支持9.2,系统cuda安装比较简单,直接安装之后把cuda软链接改成cuda9.2的路径就行了,不用卸载之前的cuda,也不影响之前版本cuda的使用。。。

改软链接方法如下:

sudo ln -s /usr/local/cuda-9.2 /usr/local/cuda

cuda 92版本的pytorch安装的方法如下:

conda install pytorch=0.4.1 torchvision cuda92 -c pytorch

如果还需要装cuda的话:

conda install cudnn=7.3.1=cuda9.2_0

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值