分类：配置Pytorch

2021
06-04

win10系统配置GPU版本Pytorch的详细教程

一、安装cuda1、在英伟达官网下载最新版的cuda驱动https://developer.nvidia.com/zh-cn/cuda-downloads都选上就行了，然后一路默认安装输入nvcc-V查看是否安装成功二、安装pycuda1、在控制台中输入pipinstallpycuda安装pycuda2、在环境变量中添加cl.exe3、测试pycuda是否正常运行importpycuda.driverasdrvimportpycuda.tools,pycuda.autoinit,numpyimportnumpy.linalgaslafrompycuda.compilerimportSourceModulemod... 继续阅读 >