ubuntu16.04+cuda9.0+cudnn+anaconda3 安装pytorch

安装cuda

参考博客 https://blog.youkuaiyun.com/weixin_42795611/article/details/93065567  写的很详细

安装anaconda

首先下载anaconda,由于官网的下载速度十分缓慢,而且还不支持断点续传,经常下载失败。故而,去国内的镜像站点下载,

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

 选择对应的版本下载,下载完成后上传到linux机器上去,服务器选择filezilla上传,虚拟机直接拖拽进去就ok了。

然后打开终端,输入:

bash Anaconda3-5.2.0-Linux-x86_64.sh

一路回车+输入yes即可。

完成安装后,关闭终端,并重新打开一个终端,输入conda:

则安装完成,输入:sudo gedit ~/.bashrc

sudo gedit ~/.bashrc

检查末尾是否已经添加如下变量,没有则添加:

 
2.创建虚拟环境 

   conda create -n name python=3.5 (name和python版本号可以更改)

3. 进入虚拟环境

   source activate name 

4. 安装pytorch

 conda install pytorch=0.4.0 torchvision cudatoolkit=9.0(安装指定版本的pytorch)

注:如有需要可以将默认的源换成国内的清华源,由于一些原因默认的源国内下载速度过慢,在终端输入以下:
//这个是清华pytorch源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
查看修改是否成功,输入:
 
cat ~/.condarc

安装好后可通过 conda info --envs 命令查看当前已安装的pythorch个数和名称

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值