Conda配置pytorch

目录

一、conda创建虚拟环境(环境名称为FCN,可随意更改)

二、激活虚拟环境

三、查看cuda版本

四、安装Pytorch

五、测试pytorch是否安装成功

一、conda创建虚拟环境(环境名称为FCN,可随意更改)

conda create -n FCN python==3.8

二、激活虚拟环境

刚刚创建了FCN的虚拟环境,之后的操作均在FCN下进行

在命令行输入如下代码进入环境:

conda activate FCN

三、查看cuda版本

1.在命令行输入nvidia-smi,红框中展示的为你的cuda版本

四、安装Pytorch

1.查看与自己电脑系统对应的pytorch版本

pytorch官网(https://pytorch.org/get-started/previous-versions/)可以查看不同pytorch版本在不同操作系统的安装方式。以下为示例(可以根据自己的系统和cuda版本进行选择):

2.我的cuda版本是11.4,系统是Lunix

目前最新的pytorch版本是1.12.0,我选择安装pytorch版本为1.11.0:

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

如果显卡不支持CUDA,也可以安装CPU版本的pytorch,例如使用:

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cpuonly -c pytorch

五、测试pytorch是否安装成功

1.在FCN环境下进入python,输入:

python

 2.进入python后继续输入:

import torch
import torchvision
torch.cuda.is_available()

3.如果看到如下则证明安装成功:

  • 4
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
回答: 如果你的conda环境中已经安装了其他版本的pytorch,并且你想安装1.11.0版本的pytorch,你可以使用以下命令来安装:conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch。\[1\]这个命令会安装指定版本的pytorch,并且会自动处理pytorch和cuda之间的对应关系。如果你的显卡不支持CUDA,你也可以安装CPU版本的pytorch,使用命令:conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cpuonly -c pytorch。\[3\]安装完成后,你可以在FCN环境下进入python,并输入import torch和import torchvision来测试pytorch是否安装成功。然后,你可以使用torch.cuda.is_available()来检查是否支持CUDA。如果返回True,则表示安装成功。\[3\] #### 引用[.reference_title] - *1* *3* [Conda配置pytorch](https://blog.csdn.net/baidu_38662087/article/details/127504440)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down28v1,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [conda 安装pytorch(gpu)](https://blog.csdn.net/weixin_44737691/article/details/129505437)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down28v1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小薛薛snow

如果对您有帮助,请多多鼓励

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值