赞
踩
参考博客
https://blog.csdn.net/gzygwp/article/details/99719253?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param
CUDA 和cudnn 的安装
cuda安装
cuda 10.1
驱动如果比要安装的版本新,则取消版本驱动安装。
cuda 安装成功
Cudnn安装
下载cudnn,和cuda 版本要一致
将其更名解压到cuda 安装路径下
Pytorch和torchvision 的安装
2.1 Pytorch安装
这里不太推荐按照官网给的指令下载,下载很慢,而且容易出问题
2.1.1 普通方法:
首先进入 Pytorch 官网:https://pytorch.org/
点击Get Started, 选择合适的版本
按照给出的指令输入到anaconda promopt里输入
activate pytorch
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
虽然是官方方法,但是不论用conda还是pip安装都会很慢,在此推荐第二种方法
2.1.2 镜像下载安装(强烈推荐)
首先进入此链接 torch清华镜像源链接.
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
pytorch-1.60 python 3.6 cuda 10.1 cndnn7.0 .bz
进入到安装包所在文件夹
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。