标签:cudatoolkit cudnn pytorch 版本 tensorflow gpu 安装
创建环境进入环境
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__)
验证安装结果。
标签:cudatoolkit,
cudnn,
pytorch,
版本,
tensorflow,
gpu,
安装
From: https://www.cnblogs.com/ducn/p/18015846