pytorch调用3090 Ti 对应CUDA版本为11.7
时间: 2024-06-05 22:07:15 浏览: 251
pip安装pytorch的wheel文件(CUDA11.7 python3.10)
目前(截至2021年12月)最新的CUDA版本为11.5,因此如果要使用3090 Ti,需要安装11.5或更高版本的CUDA驱动程序。同时,需要确保安装了与CUDA版本匹配的PyTorch版本,可以在PyTorch官网上查看具体的版本兼容性信息。此外,需要安装相应的cuDNN库和NCCL库,以支持深度学习计算的加速和多GPU训练。
阅读全文