ubuntu环境下安装pytorch和cuda版本不匹配问题
·
实验室服务器cuda:11.4
网上搜索conda install 和pip install命令几次都装不上，本来要装11.4对应的，但是目前清华源不支持，所以需要去装清华源支持的11.1，尝试离线安装，先去清华源网站找到对应cuda和python版本的包：https://download.pytorch.org/whl/torch_stable.html
比如我的
将需要安装的.whl包下载到服务器里
然后在相应目录的命令行安装
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple https://download.pytorch.org/whl/cu111/torch-1.10.0%2Bcu111-cp37-cp37m-linux_x86_64.whl
pip list查看安装成功否：
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
1
0- 0
已为社区贡献1条内容
所有评论(0)