1.下载Anconda
2.构建虚拟环境conda create -n 你需要的名字
3.修改镜像源(清华或者上海交大)
4.https://pytorch.org/(安装对应版本的pytorch)
(去掉-c pytorch 表示不去pytorch官网下载,到设置的镜像源去下载)注意python的版本!!!
5.添加到pycharm中
6.如果要安装tensorboard或者tensorboardx,需要先安装tensorflow
(注意tensorflow版本和python版本的对应关系)
conda create -n pytorch_py3.6 python=3.6
conda activate pytorch_py3.6
conda deactivate
nvidia-smi 查看cuda 对应的版本
使用清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
测试pytorch
import torch
torch.cuda.is_available()
torch.cuda.get_device_name(0)
不同的Python对应不同的TensorFlow(注意!!!)
安装tensorflow
pip install tensorflow-gpu==1.12
conda install tensorboard
conda install tensorboardx
pytorch的tensorboard是基于tensorflow的,因此要想使用tensorboard需要安装tensorflow