2024年2月14日

conda安装gpu版本pytorch与gpu版本tensorflow

摘要： 创建环境进入环境 nvidia-smi查看cuda版本，根据cuda版本安装对应版本的pytorch，在pytorch官网可以查看，版本不合适可以使用较低版本cuda的torch，使用官网提供的命令行安装即可，import torch``print(torch.cuda.is_available() 阅读全文

posted @ 2024-02-14 23:19 章序 阅读(131) 评论(0) 推荐(0) 编辑