3050ti安装TensorFlow与Pytorch

电脑环境:在这里插入图片描述472.47
在这里插入图片描述11版本cuda

1.首先在已有的anaconda环境下建立新的虚拟环境。
conda create -n third_env python=3.7
conda activate third_env

2.然后下载cuda11.2以及对应的cudnn,
安装cuda11.2,自定义,不勾选visual studio
其他默认下一步安装
安装检查:
nvcc -V
出现cuda 11.2即安装成功cuda

3.接着将cudnn解压后复制到v11.2目录下
配置环境变量-系统变量
在这里插入图片描述
即安装好cudnn

4.TensorFlow安装,在建好的虚拟环境下,输入命令:
pip install tensorflow-gpu==2.5

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
安装完成
5.新建环境,安装Pytorch
启动虚拟环境后,输入:
pip3 install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
在这里插入图片描述

安装成功后,进入python环境,进行测试即可。
安装感受:在网好的情况下安装,确认好电脑版本,要下载的cuda版本,对应的TensorFlow版本,Pytorch版本再安装,不然会出现版本不匹配问题。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值