记录一个小坑
在学习深度学习时我们一般会在github上down大佬们的代码,
然后用命令pip install -U -r requirements.txt
来一次性安装所需要的依赖环境,但有时系统会理解错,比如requirements里面写的是torch>=1.7.0 torchvision>=0.8.1,pip自动安装的是cpu版本,在cmd命令行用 torch.cuda.is_availiable()
命令发现问题不是gpu版本
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AttributeError: module 'torch.cuda' has no attribute 'is_availiable'
故需要在https://pytorch.org/get-started/locally/另外下。
当然建议你在跑别人代码时最好先配置好基础环境,再去pip install第三方的依赖包。