新建环境命令,以安装pytorch为例
conda create -n pytorch_gpu python=3.6
这里面的‘pytorch_gpu’是环境名字,可以自己随便取
进入环境命令
activate pytorch_gpu
添加kernel
pip install ipykernel
python -m ipykernel install --name pytorch_gpu
这里面的‘pytorch_gpu’是核心的名字,也可以随便取
安装tf-gpu
pip install --upgrade --ignore-installed tensorflow-gpu