ubuntu安装pytorch

最新推荐文章于 2024-05-08 22:28:48 发布

W.S.Yin

最新推荐文章于 2024-05-08 22:28:48 发布

阅读量256

点赞数

本文链接：https://blog.csdn.net/qq_36679208/article/details/105070650

版权

安装cuda

注意和系统内核对应，例如2080ti,可能安装cuda10.1 (430,418),如果不对应，你可能输入nvcc- V可以显示CUDA信息,但是输入nvidia-smi不显示显卡驱动信息了，也就错了

去官网注册个帐号，下载和CUDA对应的cudnn，然后解压，把文件链接到cuda就可以了，具体操作看其他文章

注意，安装时如果要安装Anaconda，则安装opencv一定要在前，否则在make -j8步骤会出错;如果已经安装了Anaconda，你可以先注释掉Anaconda环境变量，再安装opencv;
你也可能会遇到/sbin/ldconfig.real: /usr/local/lib/libcudnn.so.7 不是符号链接解决办法：建立软连接sudo ln -sf /usr/local/cuda-10.1/lib64/libcudnn.so.7.1.3 /usr/local/cuda-10.1/lib64/libcudnn.so.7;
如果你想要在Anaconda 的envs里安装opencv, 你可以在这个环境下输入conda install -c menpo opencv; 否则你可能会安装到系统python下

关注