torch cuda 安装
时间: 2023-11-07 13:04:33 浏览: 88
根据提供的引用内容,安装torch和cuda可以按照以下步骤进行:
1. 首先,查询可支持的最高cuda版本,并查看cuda、cudnn、pytorch、torchvision对应版本。
2. 根据上述查询结果,从官网下载并安装对应版本的cuda和cudnn。
3. 安装指定版本的pytorch、torchvision、cudatoolkit。在命令行中输入以下指令并加上版本号:
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
4. 如果输出为True,则代表安装成功。
需要注意的是,安装cuda和cudnn时最好按照官方教程进行,这样成功率会更高。直接使用命令行安装有较大的失败风险。
阅读全文