本教程未写完
查看显卡算力
https://developer.nvidia.cn/zh-cn/cuda-gpus
下载CUDA
https://developer.nvidia.com/cuda-toolkit-archive
下载CUDNN
https://developer.nvidia.cn/rdp/cudnn-archive
安装视频
https://www.bilibili.com/video/BV1dd4y1k7Ru/?spm_id_from=333.788.recommend_more_video.-1&vd_source=de0d211161a3b451f44a5c8e732dcbe2
https://www.bilibili.com/video/BV1nL4y1b7oT/?spm_id_from=333.337.search-card.all.click&vd_source=de0d211161a3b451f44a5c8e732dcbe2
国内镜像由于大小问题都是cpu版,参考
https://blog.csdn.net/weixin_42838061/article/details/113107234?utm_source=app&app_version=5.0.1&code=app_1562916241&uLinkId=usr1mkqgl919blen
Anaconda下载
https://www.anaconda.com/products/individual
为pytorch准备工作空间
- 打开Anaconda Prompt,执行命令
这里的pytorch可以是任意名字
conda create -n pytorch python=3.8
激活为pytorch准备的工作空间
conda activate pytorch
查看包列表
pip list
安装pytorch
- 官网:https://pytorch.org/get-started/locally/
- 可以选择需要的版本,生成安装命令
- 之后可以在查看pip list,观察pytorch是否安装成功
conda install pytorch torchvision torchaudio cpuonly -c pytorch-lts
安装编辑器PyCharm
- 安装社区版即可
https://www.jetbrains.com/pycharm/
设置PyCharm编译环境
-
选择私有环境
-
找到Anaconda安装包,如图配置(我这里由IDE工具自动填充了内容)
安装jupyter
- 给pytorch工作空间安装jupyter(base空间默认装好的)
conda activate pytorch #切换工作空间,如果已在可不操作
conda install nb_conda
启动jupyter
jupyter notebook
新建notebook
- 选择pytorch工作空间
测试notebook运行
运行快捷键Shift+Enter
返回false不支持显卡加速,不影响使用
import torch
torch.cuda.is_available()
安装tensorboard图表工具
pip install tensorboard -i https://pypi.douban.com/simple/
生成tensorboard样例数据并查看
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")
for i in range(100):
writer.add_scalar("y=x",i,i)
writer.close()
在Terminal中运行tensorboard
注意路径是logs上一层
tensorboard --logdir logs
---------------------常用命令
#装conda切换源
https://zhuanlan.zhihu.com/p/449701244
#创建虚拟环境
conda create -n py38 python=3.8
ssl报错报错看这个
https://blog.csdn.net/xiangfengl/article/details/127597065
切换环境报错解决方法:
修改环境变量----->C:\ProgramData\Anaconda3\Scripts (必须删除,不能保留,否则报错)
1、改成 C:\ProgramData\Anaconda3\condabin
2、以及C:\ProgramData\Anaconda3
conda activate py38
pip install -r .\requirements1.txt
-------------------报错集合---------------------
pytorch版本过高的修改
AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘
https://blog.csdn.net/Thebest_jack/article/details/124723687
报错 RuntimeError: “slow_conv_transpose2d_out_cpu“ not implemented for ‘Half‘
将model.half()和img.half()改为.float()
module ‘numpy’ has no attribute “float” 报错
pip uninstall numpy #删除
pip install numpy==1.23.5 #指定版本安装
#cv2找不到,引入代码改成
import cv2
------------vscode切换python源--------------
按键Ctrl+Shift+P
输入
Python:Select Interpreter
-------------conda常用-----------------
pytorch仓库的创建
#创建虚拟环境
conda create -n your_env_name python=3.8
#删除虚拟环境
conda remove -n your_env_name(虚拟环境名称) --all
#激活虚拟环境
conda activate your_env_name
#退出虚拟环境
conda deactivate
#查看列表
conda env list
#查看显卡
nvidia-smi