- 安装开源代码的环境的时候,需要安装torch时,不能简单地通过
pip install -r requirements.txt
安装
这样会导致,pytorch版本正确(都是pytorch 1.10.0),但是cuda版本对应不上的问题
cuda版本查看方法nvidia-smi
或者nvcc -V
。两种方法查出的cuda版本不一致的原因,参考https://blog.csdn.net/sophicchen/article/details/120782209
- 在pytorch官网找到相应torch版本和cuda版本的安装方法。链接:https://pytorch.org/get-started/previous-versions/
根据需要选择conda
或者pip
安装
pytorch安装
最新推荐文章于 2024-09-16 16:29:28 发布