天天看点

入坑记录(WIN10安装CUDA、CUDNN、Anaconda、pytorch)

入坑记录(WIN10安装CUDA、CUDNN、Anaconda、pytorch)

环境:

visual stdio 2017

1.CUDA

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

入坑记录(WIN10安装CUDA、CUDNN、Anaconda、pytorch)

选择相应的操作系统、版本、安装类型,我的选择为windos系统64位操作系统,cuda10最新版本,本地模式

然后就是下载(玩手机、看电影、抠脚…)

下载后点击运行exe文件,安装时会让你选择一个文件夹作为抽取文件临时存放,记住这不是安装文件夹,安装文件夹系统会自动安装到C盘,临时文件夹安装完后会自动删除。

注意请选择自定义安装,并取消Visual Studio Integration的选项不然会与原生VS产生冲突

安装完成后在cmd内测试是否成功

入坑记录(WIN10安装CUDA、CUDNN、Anaconda、pytorch)

2.CUDNN

官网地址:https://developer.nvidia.com/rdp/cudnn-download

注册填写个人信息后选择cuda相应版本的cudnn

入坑记录(WIN10安装CUDA、CUDNN、Anaconda、pytorch)

将cudnn解压后,将其全部复制到下图路径中

入坑记录(WIN10安装CUDA、CUDNN、Anaconda、pytorch)

3.Anaconda

官网地址:https://www.anaconda.com/distribution/#download-section

入坑记录(WIN10安装CUDA、CUDNN、Anaconda、pytorch)

注意选择操作系统,别下错了,默认是macOS(我第一次就下错了,获得一个pkg文件嘤嘤嘤)

入坑记录(WIN10安装CUDA、CUDNN、Anaconda、pytorch)

安装的把环境变量加上(虽然他不推荐)

注意!!!!如果中途跳出cmd窗口不要手贱关掉!!!我被坑了两次!!!

4.PyTorch

官网链接:https://pytorch.org/

入坑记录(WIN10安装CUDA、CUDNN、Anaconda、pytorch)

选择相应的版本成功获得指令

进入cmd界面复制粘贴

后就是玩手机、看电影、抠脚…

入坑记录(WIN10安装CUDA、CUDNN、Anaconda、pytorch)

最后在python环境下导入torch试试看!!(我还顺便装了pytorchvision)

继续阅读