铜山网站开发,网页设计教程23,小游戏网站开发者,建设400官方网站Visual Studio配置libtorch visual Studio安装cuDNN安装CUDAToolkit安装libtorch下载Visual Studio配置libtorch(cuda版本配置) visual Studio安装
visual Studio点击安装 具体的安装和配置过程这里就不进行细讲了#xff0c;可以参考我这篇博客Visual Studio配置OpenCV(保姆… Visual Studio配置libtorch visual Studio安装cuDNN安装CUDAToolkit安装libtorch下载Visual Studio配置libtorch(cuda版本配置) visual Studio安装
visual Studio点击安装 具体的安装和配置过程这里就不进行细讲了可以参考我这篇博客Visual Studio配置OpenCV(保姆及教程) cuDNN安装
cuDNN点击安装 查看自己的显卡驱动版本以确定cuDNN的安装版本一般是最新版本的就没什么问题。 直接选择最新的进行安装,具体安装过程这里就不进行说明了。 CUDAToolkit安装
CUDAToolkit点击安装 选择配套的CUDAToolkit这里需要CUDA的12.x版本在下方红框中随便选择一个安装即可 这里将CUDAToolkit进行解压会出现以下目录将其全部复制粘贴到cuDNN所在的目录下一般默认安装位置为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.3 进入环境变量将以下路径添加至环境变量中 打开窗口【win R】输入cmd再输入nvidia-smi显示以下界面就说明CUDA安装成功了。 libtorch下载
libtorch点击下载 我们直接选择cuda版本的debug版本进行下载。需要注意的是debug和rellease版本是有区别的。 下载完成之后解压将以下路径放置环境变量中 Visual Studio配置libtorch(cuda版本配置) 打开属性进行配置 点击调试点击环境添加路径至环境 点击VC目录 点击包含目录 将libtorch和cuda的路径进行添加。 点击VC目录 点击库目录将libtorch和cuda路径进行添加。 点击c/c点击常规点击附加包含目录添加$(IncludePath),将SDL检查改为否 点击语言点击c语言标准改为c17 根据自己报错进行更改我的是c17默认为c14.将符合模式改为否 点击链接器点击输入点击附加依赖项将你安装的libtorch路径E:\DownloadTemp\libtorch\lib下的所有.lib文件进行添加。 到这里你的libtorch的cpu版本就已经配置完成了如果需要配置gpu版本就跟以下这个链接器的命令行有关系在命令行的其他选型中输入/INCLUDE:?ignore_this_library_placeholderYAHXZ 不同的CUDA版本需要不同的命令行根据自己的版本进行配置。 输入以下代码进行测试 #include iostream
#include vector
#includetorch/torch.h
#includetorch/script.husing namespace std;int main() {auto a torch::rand({ 5, 3 });if (torch::cuda::is_available()) {cout cuda可用 endl;}else{cout 不可用 endl;}cout a endl;
}显示以下界面说明你的环境以配置成功
