anoconda下安装pytorch

参考博客

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
anoconda下安装pytorch_第1张图片
驱动如果比要安装的版本新,则取消版本驱动安装。
anoconda下安装pytorch_第2张图片
anoconda下安装pytorch_第3张图片
cuda 安装成功
anoconda下安装pytorch_第4张图片

Cudnn安装
下载cudnn,和cuda 版本要一致

anoconda下安装pytorch_第5张图片

anoconda下安装pytorch_第6张图片

将其更名解压到cuda 安装路径下
anoconda下安装pytorch_第7张图片

Pytorch和torchvision 的安装
2.1 Pytorch安装
这里不太推荐按照官网给的指令下载,下载很慢,而且容易出问题
2.1.1 普通方法:
首先进入 Pytorch 官网:https://pytorch.org/
点击Get Started, 选择合适的版本
anoconda下安装pytorch_第8张图片
按照给出的指令输入到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
anoconda下安装pytorch_第9张图片
进入到安装包所在文件夹
anoconda下安装pytorch_第10张图片

你可能感兴趣的:(神经网络和深度学习)