Проверьте версию CUDA
nvidia-smi
nvcc --version
Upset Unicorn
nvidia-smi
nvcc --version
nvcc --version
conda list cudatoolkit
nvcc --version