最近安装torch版本的时候总遇到不兼容CUDA12.1的,所以要在同一个主机上安装不同版本的CUDA,然后修改CUDA_HOME就能实现切换,找了一些参考,还没有实践,留个坑,到时候填完了整理到里面:
参考:
非root用户安装cuda11.7教程
基于RTX3090的Ubuntu22.04安装cuda11.7和cudnn8.6
最近安装torch版本的时候总遇到不兼容CUDA12.1的,所以要在同一个主机上安装不同版本的CUDA,然后修改CUDA_HOME就能实现切换,找了一些参考,还没有实践,留个坑,到时候填完了整理到里面:
参考:
非root用户安装cuda11.7教程
基于RTX3090的Ubuntu22.04安装cuda11.7和cudnn8.6