【配置】Ubuntu上配置pytorch图文全流程(包括探索过程)

本文详细记录了在Ubuntu上配置PyTorch的全过程,包括使用conda创建虚拟环境,激活虚拟环境,解决CUDA版本不匹配问题,从NVIDIA官网下载并安装合适的CUDA版本,以及在遇到问题时的探索与解决方法。在安装过程中,作者发现CUDA版本为11.6,而PyTorch官网最高支持11.3,最后成功安装了cuda-11.6并完成了PyTorch的安装和测试。
摘要由CSDN通过智能技术生成


本文旨在记录在实验室服务器上配置pytorch的全过程

预备

使用conda创建虚拟环境

首先使用命令conda env list查看当前存在的环境:
在这里插入图片描述
然后使用指定python版本创建虚拟环境,使用conda create -n torchEnv python=3.9,这里的-n为虚拟环境名字参数,这里为pytorchEnv,指定python版本为3.9;安装完成后,再次使用conda env list查看当前存在的环境,其中的*号表示当前的虚拟环境:
在这里插入图片描述

激活(切换)虚拟环境

由于收到Window上安装使用pytorch的影响,所以一直在使用activate pytorchEnv命令来激活创建好的虚拟环境,但是一直没有出现任何反应,查询到linux上应该是使用source activate torchEnv来激活,这时候,激活成功后,命令行的用户名前多了(torchEnv):

  • 3
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值