一、pytorch安装
折腾了一潘~~~写了一个总结。
1.首先看自己的笔记本有gpu不,有就去更新独立显卡驱动
驱动程序更新
NVIDIA 驱动程序下载www.nvidia.cn如果没有就直接跳到安装pytorch
2.安装驱动后,然后看自己笔记本所支持的cuda的版本
命令:
nvcc -V 和nvidia-smi
3、下载对应的cuda,安装。这里注意:选择自定义安装,而重复的软件就不用安装
链接:
CUDA Toolkit Archivedeveloper.nvidia.com4、然后windows+r快捷键,输入cmd,输入命令nvcc -V,如果出现下面字样则说明成功安装
5,安装对应的cudnn,将其文件对应的文件夹替换cuda的文件
链接:
https://developer.nvidia.com/rdp/cudnn-archivedeveloper.nvidia.com6,添加清华镜像
链接:
Anaconda 查看、添加、删除 安装源blog.csdn.net注意:查看当前镜像:conda config --show-sources
7、最后安装pytorch就好(安装以前版本的命令)
conda install pytorch torchvision cudatoolkit=9.0 #只需把9.0改了就好
注意:这个命令有待研究,不一定都准确
出现错误: E:app4anaconda3pkgspytorch-1.5.1-py3.6_cuda92_cudnn7_0.tar.bz2. 删除文件(不是文件夹)
8、测试是否正确安装
import torch
torch.__version__ # 返回版本号
torch.cuda.is_available() # 返回True
二、ubuntu安装---
1、下载虚拟机
2、下载ubuntu ios文件、
3、在ubuntu安装pytorch待续--------------
注意:一开始的屏幕有点小,我们要调整ubuntu屏幕---
xrandr
xrandr -s 1600x1200