离线安装pytorch

最新推荐文章于 2024-09-17 15:39:48 发布

MarsGril

最新推荐文章于 2024-09-17 15:39:48 发布

阅读量1.2k

点赞数

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/wei_yanwen/article/details/132203275

版权

本文详细描述了如何在离线环境下确认CUDA版本，下载并安装PyTorchwheel，处理gcc版本过低的问题，以及如何正确配置Python和相关包，如torchvision和torchaudio，同时提供了解决gcc库路径问题的方法。

摘要由CSDN通过智能技术生成

离线安装pytorch

文章目录

离线安装pytorch

确认cuda版本

nvidia-smi
ncvv -V
找到一个可兼容的最低版，
比如cuda是11.2的版本，pytorch可支持的最低版本是11.1

下载pytorch的wheel

在官网发现，下载支持cuda的pytroch离线版，可以查看这个网址download-pytorch。
https://download.pytorch.org/whl/cu111
（把给定的cu118改成cu111即可）

仔细在里面找到了cu111下跟我环境一致的wheel
注意暂时还不支持py311（现在是2023年8月），所以，我把原来安装的Python-3.11删掉重新安装了Python-3.9，
方法是下载一个2022年10月份左右的Anaconda。
目前2023年8月的anaconda是默认的3.11版本的python，离线安装再建一个env比较麻烦，所以直接下载对应Python版本的conda就好。
tuna
（https://mirrors.tuna.tsinghua.edu.cn/anaconda/）

一个潜在的问题是，pip安装的时候会WARNING无法升级最新版（因为离线），但是不影响目前版本的pytorch安装。

安装torch及其支持包

根据官网的建议，先安装torch，再安装torchcision和torchaudio，个人感觉后面两个不需要的话可以不装。

pip install torch-1.10.0+cu111-cp39-cp39-linux_x86_64.whl

后续：gcc版本太低了，出现一个错误

很多教程都是教把ananconda3里面的cxxabi放到/usr/lib里面去，其实还有一个更好解决方案，尤其是没有root权限的情况下，直接把anaconda3的lib链接在LD_LIBRARY_PATH里即可。
不删除文件方法
（参考：https://blog.csdn.net/tlm414/article/details/54914916?spm=1001.2101.3001.6650.8&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ERate-8-54914916-blog-77854581.235%5Ev38%5Epc_relevant_sort_base1&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ERate-8-54914916-blog-77854581.235%5Ev38%5Epc_relevant_sort_base1&utm_relevant_index=15
）