cuda10.0安装torch.7.0
时间: 2023-09-28 15:04:53 浏览: 106
为了在使用 CUDA 10.0 的机器上安装 Torch 7.0,需要按照以下步骤:
1. 安装 CUDA 10.0,可以从官网下载对应版本的 CUDA 安装程序。
2. 安装 cuDNN v7.4.1,也可以从官网下载并按照说明安装。
3. 安装 Anaconda,可以从官网下载对应版本的 Anaconda 安装程序。
4. 创建一个新的 Conda 环境,可以使用以下命令:
conda create --name torchenv python=3.6
5. 激活新的环境:
conda activate torchenv
6. 安装 PyTorch,可以使用以下命令:
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
7. 测试 PyTorch 是否正确安装:
python -c "import torch; print(torch.cuda.is_available())"
如果输出 true,代表 PyTorch 安装成功。
注意:在安装过程中,需要根据自己的环境和需求选择正确的版本,安装命令中的版本号也需要对应。
此外,建议在安装 PyTorch 之前给 Anaconda 和 conda 更新一下:
conda update conda
conda update anaconda
阅读全文