Cannot train Tao Toolkit UNet model in version v4.0.0 and v4.0.1
![Cannot train Tao Toolkit UNet model in version v4.0.0 and v4.0.1](https://global.discourse-cdn.com/nvidia/optimized/4X/e/8/f/e8f7aa7708574c0834b3f8b8b628f695be85488d_2_650x500.png)
Excuse me @Bin_Zhao_NV @Morganh I’ve changed gpus from Tesla P100 to Tesla V100 and tried to train Tao Toolkit UNet model with 4 gpus in version v4.0.0 and v4.0.1 again. However. I still got the error message: device CUDA:0 not supported by XLA service while setting up XLA_GPU_JIT device number 0. This is the result in the process of training UNet when I ran the command nvidia-smi. Is this a bug for Tao Toolkit v4.0.0 and v4.0.1 ? When I trained UNet in the version v3.22.05, it seemed tha
![](https://www.groove3.com/images/g3plus/2024-AAP-400x.png)
Access the Entire Tutorial Library
![](https://www.drware.com/wp-content/uploads/2021/07/fb_image-521.png)
Train smarter with NVIDIA pre-trained models and TAO Transfer Learning Toolkit on Microsoft Azure - Microsoft Community Hub
![](https://read.nxtbook.com/ieee/vehicular_technology/vehiculartechnology_dec_2023/assets/8bcaec041acf3e63140c820fe977dbc3.jpg)
Digital Twin and Artificial Intelligence-Empowered Panoramic Video Streaming
![](https://www.digikey.com/htmldatasheets/production/870392/0/0/1/media/bg1.png)
SFM Series Drawing, Double Row by Samtec Inc. Datasheet
![](https://www.researchgate.net/publication/367051480/figure/fig5/AS:11431281222510558@1707303559720/Performance-comparison-between-IML-TYLCVs-and-the-top-5-baseline-models-on-training.jpg)
Performance comparison between IML-TYLCVs and the top 5 baseline models
![](https://flow3d.co.kr/wp-content/uploads/Figure-1-Schematic-of-the-present-research-model-with-dimensions-and-macro-roughnesses-installed.png)
Foam FLOW-3D
![](https://chemistry-europe.onlinelibrary.wiley.com/cms/asset/05a830d9-f457-4fc9-b154-602cb431c744/cphc201700665-toc-0001-m.png)
Molecular Catalysts for N2 Reduction: State of the Art, Mechanism, and Challenges - Roux - 2017 - ChemPhysChem - Wiley Online Library
![](https://pub.mdpi-res.com/remotesensing/remotesensing-15-01684/article_deploy/html/images/remotesensing-15-01684-ag.png?1679976552)
Remote Sensing, Free Full-Text
![](https://read.nxtbook.com/ieee/signal_processing/signal_processing_jan_2023/assets/fe2b8a45c48f10c9d1fb072abe03523f.jpg)
Imaging With Equivariant Deep Learning
![](https://www.nxp.com/assets/images/en/dev-board-image/MR-CANHUBK344-BOTTOM.png)
S32K344 Evaluation Board for Mobile Robotics with 100BASE-T1 and Six CANFD
![](https://gallery.streamlinevrs.com/units-gallery/00/05/C1/image_165551635.jpeg)
Roelens Vacation Rentals: Salty Shoreline, Cape Coral in Florida – Roelens Vacation Rentals
![](https://www.mdpi.com/land/land-12-01602/article_deploy/html/images/land-12-01602-g001.png)
Land, Free Full-Text
![](https://cdn-prod.scdn6.secure.raxcdn.com/static/media/d3ef0b351ed642f78f48f23821b245bf.jpg?_cb=1649427908.3992825)
Training the RetinaNet or ResNet18 model with TAO Toolkit and deploying it with DeepStream and Triton, Developing and Deploying Vision AI with Dell and NVIDIA Metropolis
![](https://global.discourse-cdn.com/nvidia/original/3X/4/d/4d64cbe309f0fc43d14502905b44fa316f922b36.png)
The training process of Tao-Toolkit-API unet is always in Inf status - TAO Toolkit - NVIDIA Developer Forums