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

你可能感兴趣的:(RTX2080显卡与pytorch兼容问题:cuda9.2)