记录在ubuntu RTX2080Ti下配置torch

配置成功时的环境:

ubuntu 16.4 + RTX 2080Ti + nvidia driver 410.78 + cuda 10.0 + cudnn 7.5 + cmake 3.13

./install.sh 中遇到的问题:

1. nvcc fatal   : Unsupported gpu architecture 'compute_75'

是nvidia driver的版本和cuda的版本不匹配造成的。之前用cuda 9.0版本太低,后换成cuda 10.0解决。nvidia driver 和cuda版本的对应参考官方doc:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cuda-libraries-knonwn-issues

2. error: cannot overload functions distinguished by return type alone

在cuda 10.1是碰到,可以参考https://github.com/torch/cutorch/issues/834,按照nagadomi的方法解决了这个问题,但是又报其他错误。改成cuda 10.0后没有报这个错误,可能还是cuda和nvidia driver版本不匹配的锅??

3. cmake安装

扫描二维码关注公众号,回复: 5750171 查看本文章

解压文件并进入文件目录,然后执行

./bootstrap
make -j4
sudo make install

猜你喜欢

转载自blog.csdn.net/britney_f/article/details/88971830
今日推荐