Ubuntu系统下搭建pytorch1.2gpu版本的环境

该博客详细介绍了如何在服务器上下载并安装Anaconda,包括选择安装路径、添加环境变量等步骤。接着,创建了一个名为pytorch1.2的conda环境,并安装了pytorch与cudatoolkit。最后,检查了pytorch的GPU版本是否安装成功,整个过程详细且易懂。
摘要由CSDN通过智能技术生成

1.Anaconda 的安装

首先在百度网盘中下载安装包链接:https://pan.baidu.com/s/10WF8a9dzLNztcCDU2VaF7w 
提取码:1234

http://链接:https://pan.baidu.com/s/10WF8a9dzLNztcCDU2VaF7w 提取码:1234

我的是2021五月份的版本

 下载完成以后,首先用Xftp上传到服务器当中,我是直接上传到了根目录下,然后在服务器命令行当中进入Anaconda安装包所在的目录当中使用如下命令安装

bash Anaconda3-2021.05-Linux-x86_64.sh

然后一直按回车键,会出现如下界面,输入yes,按回车

 

Do you accept the license terms? [yes|no]
[no] >>> 
Please answer 'yes' or 'no':'
>>> yes

 出来这个页面是提示你让你输入anaconda的安装路径,

Anaconda3 will now be installed into this location:
/root/anaconda3

  - Press ENTER to confirm the location
  - Press CTRL-C to abort the installation
  - Or specify a different location below

[/root/anaconda3] >>> 

改变安装目录,输入要改变的安装目录(注意不要预先新建文件夹,要空文件夹),我是将其安装

在了/root/anaconda3目录下

添加环境变量,选yes

至此出现下面的页面,anaconda算是安装完成了 

然后输入conda命令进行测试, 

结果出现-bash: conda: command not found,这个时候进入root目录,输入source ~/.bashrc 进行配置文件的激活,之后就可以使用了。 

 

如果还是显示command not found,可以使用下面的命令行导入环境变量(/root/anaconda3/bin中间是路径名,/root/anaconda3是anaconda安装文件夹,./bin是其安装目录) 

export PATH=$PATH:/root/anaconda3/bin

下面进行pytorch环境配置

使用conda命令创建一个python版本为3.6的名字为pytorch1.2

 conda activate pytorch1.2

环境:

conda create -n pytorch1.2 python=3.6

如下图所示: 

然后出现如下界面,选y,进行下载

 

创建完成之后,通过下面命令激活并进入pytorch环境:

 

 conda activate pytorch1.2

进入环境之后可以通过conda list命令查看当前已经安装的包。

source deactivate

 

CUDA和CUDNN安装

这两个服务器上都已经有,不用管,我选的镜像是cuda10.0版本的镜像

下面进行最后一步pytorch1.2版本的安装

使用下面命令激活第一步添加的pytorch 环境

conda activate pytorch

在pytorch官网  查看自己的版本对应的安装命令

在这里我安装的是cuda10.0,对应的pytorch版本,因此我的命令为

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

安装的过程会自动选择一些依赖包,建议选择yes进行安装。不断下一步即可。
安装成功后可在pytorch虚拟环境下,进入python环境,导入torch 包,如果没有错误的话,说明安装成功。

 

然后输入torch.__version__查看torch的版本

 

 

然后输入torch.cuda.is_available(),如果返回true则说明pytorch1.2GPU版本的pytorch安装成功

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值