1.如何在jupyter中切换环境
1.安装nb_conda
- 在Anaconda终端 base环境下中输入conda insatll nb_conda
2.激活对应的环境并安装ipykernel
- activate pytorch-gpu
- conda install ipykernel
- #python -m ipykernel install --user --name your_env_name
2.查看tensorflow-gpu可用的GPU
1.在python文件,输入
from tensorflow.python.client import device_lib #查看电脑的gpu
device_lib.list_local_devices()
3.tensorflow选择gpu设备
import os #使用指定GPU
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
4.安装pytorch-gpu版
尝试过conda安装方法,失败了。
看到一篇博客,通过pip,从清华镜像站方式安装。
1.首先安装CUDA和cuDNN
2.打开pytorch官网,选择pip安装,复制命令
3.将内容改为
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch===1.3.1 torchvision===0.4.2 -f https://download.pytorch.org/whl/torch_stable.html
4.安装成功