好久前学下pytorch的使用,整理下发个系列出来
1、安装 conda 环境及 pytorch环境
把需要用的环境安装到位,python的虚拟环境以及pytorch环境
#安装
conda conda create -n pytorch python=3.6#pytorch官网查找安装命令
https://pytorch.org/
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia#测试
import torch
torch.cuda.is_available()
2、安装pycharm
cd pycharm-community-2023.1.2/bin
sh ./pycharm.sh
安装pycharm 用vsode也行 看自己开发哪个方便
3、安装jupyter
我个人是比较喜欢用终端调试 jupyter也还行吧
conda install nb_conda #启动jupyterjupyter notebookshift 回车其他操作:dir() 打开 看见
help()说明书
4、一些其他需要的工具
注意虚拟环境安装包位置 安装opencv-python
python 3.9 pip3 install opencv-python -i https://pypi.doubanio.com/simple/ --target=/home/qhr/anaconda3/envs/pytorch/lib/python3.8/site-packages#安装tensorboardpip3 install tensorboard -i https://pypi.doubanio.com/simple/ --target=/home/qhr/anaconda3/envs/pytorch/lib/python3.8/site-packages
以上环境说简单也简单 遇到环境依赖问题也挺愁人
但是虚拟环境会用的话 问题不大 Just like that