search

dusty-nv / jetson-containers

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T
MIT License
2.09k stars 435 forks source link

Doubt build pytorch #329

Closed johnnynunez closed 9 months ago

johnnynunez commented 9 months ago

Should we use export USE_CUDNN=1 to build Pytorch? @dusty-nv

dusty-nv commented 9 months ago

@johnnynunez cuDNN should be automatically detected by PyTorch and USE_CUDNN=1 isn't explicitly needed.

If you're on JetPack 4, make sure the default docker runtime is nvidia and DOCKER_BUILDKIT=0

johnnynunez commented 9 months ago

thanks @dusty-nv