PyTorch + Conda

1 安装 conda

在这里插入图片描述

2 在 conda 中创建python的虚拟环境

  • 可以在conda命令行中创建,也可以在conda的交互界面中创建
  • 推荐在conda交互界面创建
    在这里插入图片描述
  • 双击第一个图标
    在这里插入图片描述
    在这里插入图片描述
  • 可以方便选择环境的python版本,我这里创建了几个不同的版本,其中 base 为默认的环境

3. 安装 torch

PyTorch 下载
在这里插入图片描述

  • 这个在 conda promot 中安装比较方便
# 进入刚刚创建的新环境
conda activate envoName
# 粘贴 pytorch 下载页面中的命令行(根据需要选择不同版本)
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

4. 使用开源项目注意事项

  • requirments.txt 文件表明该项目依赖的库,利用 pip install -r requirments.txt 指令进行安装;或根据 pycharm 自带的提示操作安装(注意,有些库名与实际不相符)
conda acticvate envoName(虚拟环境名称)
  • 在代码中添加环境变量
# 例如
import os
os.environ[" str.... "] = "True"

5. Jupyter notebook 切换不同的环境

  • 首先安装Jupyter notebook,同样可以交互界面 Anaconda navigator 中安装
  • Jupyter notebook 切换到对应的 conda 环境,要先为当前环境安装 Jupyter kernels
conda install nb_conda_kernels
  • 然后就可以在Jupyter notebook 中切换不同的环境、运行代码
    在这里插入图片描述

6. tensorboard 显示日志数据

  1. 首先要安装 tensorboard
  2. 控制台 pip install tensorboard
  3. 运行你的程序,如下
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("newlogs")  # 创建日志对象,并指定日志保存的地址为 当前目录下的 newlogs 文件夹
# 制造一些数据
for i in range(100):
    writer.add_scalar("y = 2x", 2*i, i) # 添加日志数据
writer.close() # 关闭日志对象
  1. 控制台打开 tensorboard ,指定数据源和端口页面所在端口:
  2. 查看日志数据(在指定端口查看): 控制台:tensorBoard --logdir=newlogs --port=8888
  • 注意:重新绘制的时候,需要将原来的日志文件删除,否则图表会积累上一次的数据

7. 安装32位 python

安装32位python之前先需要设置set CONDA_FORCE_32BIT=1,这样环境可以切换到32位。可以通过conda info命令查看platform是否为32位进行确认

set CONDA_FORCE_32BIT=1
conda create -n python37-32 python=3.7

猜你喜欢

转载自blog.csdn.net/CXYYL/article/details/123091579