Fix CPU/CUDA build.
Description:
The CPU packages uses the build folder meant for the CUDA version, and the CUDA package uses the build folder meant for the CPU version.
The CPU package also depends on the CUDA version of the pytorch package.
Additional info:
Introduced with 0.17.0