When running the Jenkins CI with PyTorch 2.1, we get an error that the cuBLAS context is not initialized when calling skcuda.cublas.cublasSetStream. However, it works for PyTorch 2.0, so likely something has changed between these two versions.