cuda 9.0 安装torch 0.4_pytorch+ubuntu安装总结

一、pytorch安装

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

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

驱动程序更新

NVIDIA 驱动程序下载​www.nvidia.cn
v2-1238b010777e93daf2ac4c1a319f120e_ipico.jpg

如果没有就直接跳到安装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.net
v2-26d6af21428a9f5df4777a0bf1e9229e_180x120.jpg

注意:查看当前镜像: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

v2-aea68918526bb873b34420a68df09fc6_b.jpg

二、ubuntu安装---

1、下载虚拟机

2、下载ubuntu ios文件、

3、在ubuntu安装pytorch待续--------------

v2-6c8852bbed5e1a7dabae42dece3b1741_b.jpg

注意:一开始的屏幕有点小,我们要调整ubuntu屏幕---

xrandr

xrandr -s 1600x1200

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值