当前位置：网站首页>AssertionError: CUDA unavailable, invalid device 0 requested
AssertionError: CUDA unavailable, invalid device 0 requested
2022-06-24 21:19:00 【小MarkK】
在使用yolov5训练时，提示这个错误，查看torch.cuda.is_available()得到的返回值是false
查看nvidia-smi可以看到信息
由于之前使用conda安装的torch1.9.0
这里使用pip重新安装：
pip3 install torch == 1.9.0+cu111 torchvision == 0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html全部安装后，再次执行torch.cuda.is_available()时
可以看到正确结果了
参考：AssertionError: CUDA unavailable, invalid device 0 requested_消灭BUG鸭的博客-CSDN博客_cuda unavailable
边栏推荐
- Tencent cloud wecity solution
- Transform BeanUtils to achieve list data copy gracefully
- Boutique enterprise class powerbi application pipeline deployment
- Picture rotation move zoom gradient
- Activity lifecycle
- 15. several methods of thread synchronization
- Contentresolver, get the SMS content
- 论文翻译 | RandLA-Net: Efficient Semantic Segmentation of Large-Scale Point Clouds
- [untitled]
- Bi SQL constraints
猜你喜欢
随机推荐
Why does Dell always refuse to push the ultra-thin commercial notebook to the extreme?
4 ans d'expérience de travail, 5 modes de communication Multi - thread ne peuvent pas être décrits, vous osez croire?
生态护航 云服务商挥起“英特尔大旗”
Bi-sql - join
Bi SQL drop & alter
How to store dataframe data in pandas into MySQL
为猪脸识别而进行自己数据集的构建、训练「建议收藏」
Tencent has completed the comprehensive cloud launch to build the largest cloud native practice in China
Super detailed description and derivation of convolution and deconvolution (deconvolution is also called transpose convolution and fractional step convolution)
Bi skill - judge 0 and null
Using bindservice method to pause music playing
Picture rotation move zoom gradient
Bi SQL alias
[practical series] full WiFi coverage at home
matlab 取整
卷积与转置卷积
Yasea APK Download Image
屡获大奖的界面控件开发包DevExpress v22.1官宣发布
Redis persistence
Convolution and transpose convolution