conda安装gpu版本pytorch与gpu版本tensorflow

创建环境进入环境
nvidia-smi查看cuda版本,根据cuda版本安装对应版本的pytorch,在pytorch官网可以查看,版本不合适可以使用较低版本cuda的torch,使用官网提供的命令行安装即可,import torch``print(torch.cuda.is_available())验证安装结果。
tensorflow的安装要在环境中安装cudatoolkit和cudnn,可以使用conda search 了解conda提供的cudatoolkit和cudnn版本,并做好版本对应,tensorflow/python/cudatoolkit/cudnn对应版本可以在官网查看https://tensorflow.google.cn/install/source_windows,安装顺序是cudatoolkit/cudnn/tensorflow,安装后import tensorflow as tf``print(tf.__version__)验证安装结果。

posted @   章序  阅读(150)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· 上周热点回顾(2.24-3.2)
点击右上角即可分享
微信分享提示