虚拟环境内使用pip安装torch内存爆掉

目录

• 问题描述与解决方案
• • 问题描述
  • 解决方案
  • • 确定与电脑相配的cuda版本
    • 寻找对应的`.whl`文件
    • • 打开网站
    • 下载完成后进入虚拟环境

问题描述与解决方案 问题描述

在anaconda的虚拟环境内安装torch失败后，再次运行该命令内存爆满。卸载并重新安装anaconda后仍出现此问题。

解决方案

借助.whl文件进行安装

确定与电脑相配的cuda版本

本文版本为11.0

寻找对应的.whl文件 打开网站

https://download.pytorch.org/whl/torch_stable.html.
CTRL+F

选择需要的torch下载（我的虚拟环境python=3.6，Windows系统）

下载完成后进入虚拟环境

f:进入F盘
cd + 文件所在位置进入文件夹

pip install torch-1.7.1+cu110-cp36-cp36m-win_amd64.whl

import torch

print(torch.cuda.is_available())

true即为安装成功！