
pytorch
文章平均质量分 79
蓝天居士
修齐治平,先忧后乐
展开
-
pytorch中torch.randn函数
原型:torch.randn(*sizes, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor参数: sizes (int...): a sequence of integers defining the shape of the output tensor. Can be a variable number of argumen...原创 2021-01-07 09:01:11 · 5311 阅读 · 1 评论 -
Resnet Attention Model 结构
ResidualAttentionModelCovid_92( (conv1): Sequential( (0): Conv2d(3, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False) (1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (2): ReLU(inplace=True...原创 2020-11-25 20:00:54 · 596 阅读 · 0 评论 -
pytorch中数据在CPU与GPU之间的切换
数据在CPU与GPU之间来回切换的pytorch方法:数据从CPU放到GPU,即数据从CPU到GPU的迁移,使用以下语句:data.to("cuda")数据 从GPU到CPU,使用以下语句:data.to("cpu")data通常会有两种数据类型:1. Tensor2. Moduleto函数:转换数据类型/设备1. tensor.to(*args, **kwargs)2. module.to(*args, **kwargs)举例:...原创 2020-11-07 23:31:13 · 5527 阅读 · 0 评论 -
torch.max()综合
参考:https://blog.youkuaiyun.com/LEELOVESTUDY/article/details/106521130?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLear原创 2020-11-07 21:06:38 · 361 阅读 · 0 评论 -
anaconda创建虚拟环境
0. 安装anaconda详见《anaconda下载与配置》1. 创建虚拟环境conda create -n pytorch_1.4_gpu python=3.7中间过程中会提示Procceed ([y]/n)?选择y,按回车,或者直接回车2. 激活虚拟环境conda activate pytorch_1.4_gpu...原创 2020-10-30 22:24:10 · 823 阅读 · 0 评论 -
Process finished with exit code -1073741819 (0xC0000005)的问题
环境pycharmpytorch_1.4_gpu运行程序总是出现Process finished with exit code -1073741819 (0xC0000005)。开始以为是程序那里出错了,后来把程序都封掉了,只保留几个简单的import也同样出错。最终定位是import pandas导致的问题,只要一import pandas就出现这个错误。卸载重装pandas(pandas-1.1.3),更换pandas版本(更换为pandas-1.1.2版本会提示找不到pandas)也.原创 2020-10-15 20:03:15 · 3730 阅读 · 1 评论 -
CUDA:out of memory问题
基于UCSD-AI4H/COVID-CT进行训练时,出现以下错误:RuntimeError: CUDA out of memory. Tried to allocate 98.00 MiB (GPU 0; 2.00 GiB total capacity; 1.53 GiB already allocated; 9.93 MiB free; 1.55 GiB reserved in total by PyTorch)解决方法:...原创 2020-08-19 23:14:26 · 1365 阅读 · 0 评论 -
conda搭建工作环境
1. 创建虚拟环境执行 conda create -n pytorch_1.4_gpu python=3.7 命令,创建虚拟环境(具体版本根据自己实际情况设定)://///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////Collecting package metadata (current_repodata原创 2020-08-17 22:48:57 · 637 阅读 · 0 评论 -
module ‘torch.jit‘ has no attribute ‘unused‘问题
在执行from torchvision import transforms时出现了以下错误:Traceback (most recent call last): File "D:/研究生/毕业设计/COVID19/COVID19_CT/conduct.py", line 8, in <module> from torchvision import transforms File "D:\Anaconda3\envs\pytorch_1.2_gpu\lib\site-pack...原创 2020-08-17 22:35:24 · 4132 阅读 · 0 评论 -
pytorch及相关工具的安装
pytorch安装安装步骤:1. 检查是否有合适GPU,如果有,则需要安装CUDA和CuDNN。2. CUDA和CuDNN下载和安装(非必须)。3. 下载pytorch的whl文件。4. 下载torchvision的whl文件。5. 通过pip进行安装。6. 通过命令查看pytorch版本,验证pytorch是否安装成功。登录网址:https://download.pytorch.org/whl/torch_stable.html。选择合适的版本进行下载。笔者下载的.原创 2020-08-14 12:05:06 · 442 阅读 · 0 评论