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 import tensorflow as tf``print(tf.__version__)验证安装结果。

浙公网安备 33010602011771号