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 @ 2024-02-14 23:19  章序  阅读(197)  评论(0)    收藏  举报