1.安装cuda和cudnn
本机安装的cuda版本11.0.2、cudnn版本 v8.0.5_cu11.0
Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程(图文)_ubuntu20.04安装cuda_嵌入式技术的博客-CSDN博客
CUDA卸载(runfile安装以及deb方式安装方式的卸载)_cuda runfile_a little cabbage的博客-CSDN博客
更换cuDNN版本:
Ubuntu22.04安装CUDA和cuDNN详细过程记录_ubuntu安装cuda和cudnn_有时候。的博客-CSDN博客
2.安装pytorch和Anaconda
cuda版本10.2的pytorch
cuda版本11.0的pytorch1.7.0
Python深度学习:安装Anaconda与PyTorch库(GPU版本)_哔哩哔哩_bilibili
3.安装libtorch
cuda版本11.0的libtorch v1.7.0
下载地址:
https://download.pytorch.org/libtorch/cu110/libtorch-cxx11-abi-shared-with-deps-1.7.0%2Bcu110.zip
Ubuntu16.04安装Libtorch_libtorch ubuntu_zdb呀的博客-CSDN博客
pytorch官网上下载libtorch各个版本的url_baobei0112的博客-CSDN博客
需要注意的是,安装的libtorch版本一般要高于Python环境下的PyTorch版本,否则可能会出现libtorch中无法加载pytorch导出的.pt文件.
下载完直接解压即可,在Ubuntu系统下,用户自己安装的库一般放在/usr/local/lib目录下。
sudo mv libtorch-shared-with-deps-1.6.0 /usr/local/lib/
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)