前言:因为一个报错RuntimeError: Error compiling objects for extension
没有配置cl.exe环境变量,我的应用场景是需要搞定虚拟环境变量配置
RuntimeError: Error compiling objects for extension手把手带你解决(超详细)-CSDN博客
一、虚拟环境变量配置(我感觉我是瞎写的,别试)
Conda 虚拟环境中配置环境变量 (具体来说是 Pytorch 虚拟环境调用系统 CUDA)_conda环境变量-CSDN博客
粗暴添加变量,虚拟环境崩了删除变量
conda env config vars unset 变量名 -n 环境名 # 删除指定虚拟环境下的指定变量 conda env config vars list -n 环境名 # 显示指定虚拟环境下的所有变量
终于能进来了!
二、系统环境变量配置
安装之后的 CUDA 和 cudnn 会存放在虚拟环境的 lib 文件夹 或者 include 文件夹里面。
但是我在anaconda3文件夹里搜cuda显示在D:\anaconda3\pkgs
# 虚拟环境安装CUDA
conda install cudatoolkit=10.2
验证是否安装成功,结果是true就行
import torch
print(torch.cuda.is_available())