栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

实现linux服务器上创建python环境,并安装torch

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

实现linux服务器上创建python环境,并安装torch

安装anaconda实现创建python虚拟环境

这里不着重讲解如何安装anconda,下面这个链接供大家参考:

(24条消息) 科研第一步:实现Linux服务器上创建python虚拟环境_fs1341825137的博客-CSDN博客

接下来anconda环境已经创建成功。接下来就是创建python环境并安装pytorch

步骤:
conda create -n 虚拟环境名 python=版本号

以python=3.7为例创建名字为deblur的虚拟环境

conda create -n deblur python=3.7

python环境创建之后,进行pytorch的安装。

安装pytorch之前,需要查看服务器中的cuda的版本号,因为pytorch和cuda的版本需要对应:

pytorch-cuda版本的安装:
pytorch官网版本下载
选择对应的适配,就可以使用不同的指令

使用nvcc -V查看cuda的版本号。

nvcc -V

 必须使用nvcc -V查看的cuda版本号才是正确的。

然后到pytorch官网中查看相应的版本进行安装。PyTorchhttps://pytorch.org/

 可以点击图片中圈出的链接,查看以前的版本。

 以pytorch为1.7.1,并且cuda为10.1版本安装为例:

执行命令:

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

出现如图所示的内容说明pytorch安装成功:

进入python环境查看pytorch cude是否可以使用:

python import torch torch.cuda.is_available()

 出现true说明可以使用。

 注:以上步骤,需要在联网的条件下,进行安装。否则可能会不成功。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/675178.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号