ubuntu镜像源安装pytorch_pytorch+ubuntu安装总结

一、pytorch安装

折腾了一潘~~~写了一个总结。

1.首先看自己的笔记本有gpu不,有就去更新独立显卡驱动

驱动程序更新NVIDIA 驱动程序下载​www.nvidia.cn5daae2adb5649779c841e870553647f7.png

如果没有就直接跳到安装pytorch

2.安装驱动后,然后看自己笔记本所支持的cuda的版本

命令:

nvcc -V 和nvidia-smi

3、下载对应的cuda,安装。这里注意:选择自定义安装,而重复的软件就不用安装

链接:CUDA Toolkit Archive​developer.nvidia.com

4、然后windows+r快捷键,输入cmd,输入命令nvcc -V,如果出现下面字样则说明成功安装

5,安装对应的cudnn,将其文件对应的文件夹替换cuda的文件

链接:https://developer.nvidia.com/rdp/cudnn-archive​developer.nvidia.com

6,添加清华镜像

链接:Anaconda 查看、添加、删除 安装源​blog.csdn.netubuntu镜像源安装pytorch_pytorch+ubuntu安装总结_第1张图片

注意:查看当前镜像:conda config --show-sources

7、最后安装pytorch就好(安装以前版本的命令)

conda install pytorch torchvision cudatoolkit=9.0 #只需把9.0改了就好

注意:这个命令有待研究,不一定都准确

出现错误: E:\\app4\\anaconda3\\pkgs\\pytorch-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

你可能感兴趣的:(ubuntu镜像源安装pytorch_pytorch+ubuntu安装总结)