centos7安装pytorch

centos7安装pytorch0.4.0 (别的版本也适用)

  • 创建一个新的conda环境
    创建python版本为X.X、名字为your_env_name的虚拟环境。your_env_name文件可以在Anaconda安装目录envs文件下找到。
conda create -n your_env_name python=X.X(2.7、3.6等)
  • 查看所有conda环境
conda env list
  • 查看新建的conda环境中所安装过的库
pip list
  • 去pytorch官网查找对应的pytorch版本,进行安装
    官网链接:https://pytorch.org/get-started/previous-versions/
    在这里插入图片描述

在网页下面找到自己想要安装的版本,复制命令进行安装。选择版本的时候注意自己的cuda版本,我的centos7中的cuda是10.1,但是使用conda install pytorch=0.4.1 cuda90 -c pytorch命令安装的pytorch没有问题,可以使用。
查看cuda版本命令

cat /usr/local/cuda/version.txt

在这里插入图片描述
安装torchvision

pip install torchvision==0.2.1
  • over
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值